File size: 855 Bytes
399e120
 
 
fb2e956
 
 
 
 
 
 
 
 
 
399e120
fb2e956
399e120
 
fb2e956
 
 
 
399e120
fb2e956
399e120
 
 
fb2e956
 
 
 
 
 
 
399e120
 
fb2e956
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import gradio as gr
from transformers import pipeline

# Minimal stable configuration
def load_model():
    return pipeline(
        "text-generation",
        model="distilgpt2",  # Smallest reliable model
        device="cpu",
        framework="pt"
    )

model = load_model()

def generate_text(prompt):
    try:
        response = model(
            prompt,
            max_length=50,
            temperature=0.7,
            num_return_sequences=1
        )
        return response[0]['generated_text']
    except Exception as e:
        return f"Error: {str(e)}"

# Basic interface
iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt..."),
    outputs=gr.Textbox(),
    title="Stable Text Generator",
    description="Basic but guaranteed-to-work version"
)

iface.launch(server_name="0.0.0.0")