Update app.py

app.py

@@ -2,18 +2,11 @@ import gradio as gr
 import torch  # Add this import
 from transformers import pipeline
 
-model_name = "EleutherAI/gpt-neo-2.7B"  # Adjusted to use a smaller model
-generator = pipeline("text-generation", model=model_name, torch_dtype=torch.float16)
+model_name = "EleutherAI/gpt-neo-125M"
+generator = pipeline("text-generation", model=model_name)
 
 def generate_text(prompt):
-    return generator(prompt, max_length=50)[0]["generated_text"]
-
-interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
-interface.launch()
-
-
-def generate_text(prompt):
-    return generator(prompt, max_length=50)[0]["generated_text"]
+    return generator(prompt, max_length=20)[0]["generated_text"]
 
 interface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
 interface.launch()