lasting

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the text generation model
 model_id = "gpt2"  # You can replace this with any model of your choice
 generator = pipeline("text-generation", model=model_id)
 
@@ -11,15 +10,6 @@ def generate_text(prompt):
     generated_text = response[0]['generated_text']
     return generated_text
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=generate_text,
-    input=gr.inputs.Textbox(lines=2, placeholder="Enter your prompt here..."),
-    output="text",
-    title="Text Generation",
-    description="Enter a prompt and the model will generate text based on it."
-)
+demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
+demo.launch()
 
-# Launch the interface
-if __name__ == "__main__":
-    iface.launch()