Update app.py

app.py

@@ -5,10 +5,18 @@ from transformers import pipeline
 generator = pipeline('text-generation', model='gpt2')
 
 def generate_text(prompt):
-    # Generate text with a maximum length of 50 tokens
-    generated = generator(prompt, max_length=50, num_return_sequences=1)
+    # Adjust temperature to make output more focused
+    generated = generator(
+        prompt,
+        max_length=50,
+        num_return_sequences=1,
+        temperature=0.2,  # Lower temperature for less randomness
+        top_k=50,         # Optional: limit the number of choices
+        top_p=0.95        # Optional: nucleus sampling
+    )
     return generated[0]['generated_text']
 
+
 # Create a Gradio interface with one text input and one text output
 iface = gr.Interface(
     fn=generate_text,