Update app.py

app.py

@@ -1,24 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the GPT-2 model . 
-generator = pipeline('text-generation', model='gpt2')
+# Load a pre-trained Hugging Face model
+generator = pipeline("text-generation", model="gpt2")
 
-# Function to generate text
-def generate_text(prompt, max_length=50):
-    return generator(prompt, max_length=max_length, num_return_sequences=1)[0]['generated_text']
+def generate_text(prompt):
+    # Generate text using the model
+    result = generator(prompt, max_length=50, num_return_sequences=1)
+    return result[0]["generated_text"]
 
 # Create the Gradio interface
 iface = gr.Interface(
     fn=generate_text,
-    inputs=[
-        gr.inputs.Textbox(lines=2, placeholder="Enter your prompt here..."),
-        gr.inputs.Slider(minimum=10, maximum=200, default=50, label="Max Length")
-    ],
-    outputs="text",
-    title="GPT-2 Text Generator",
-    description="Generate text using the GPT-2 model."
+    inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+    outputs=gr.Textbox()
 )
 
-# Launch the app
+# Launch the interface
 iface.launch()