Update app.py

app.py

@@ -2,11 +2,11 @@ from transformers import pipeline
 import gradio as gr
 
 # Load GPT2 model
-generator = pipeline("text-generation", model="gpt2")
+generator = pipeline("text-generation", model="distilgpt2")
 
 # Function to generate reply
 def generate_reply(user_input):
-    output = generator(user_input, max_length=50, num_return_sequences=1)
+    output = generator(user_input, max_length=50, num_return_sequences=1, do_sample=True, temperature=0.8)
     return output[0]["generated_text"]
 
 # Gradio Interface