Update app.py

app.py

@@ -6,14 +6,19 @@ chatbot = pipeline("text-generation", model="gpt2")
 
 def get_response(user_input, mode="honest"):
     if mode == "honest":
-        # Regular chatbot response
-        response = chatbot(user_input, max_length=50, do_sample=True, top_k=50)
+        response = chatbot(
+            user_input,
+            max_length=100,  # Increase the length to allow complete responses
+            do_sample=True,
+            top_k=50,
+            top_p=0.95,  # More natural responses
+            temperature=0.7  # Avoid too much randomness
+        )
         return response[0]["generated_text"]
-    
+
     elif mode == "echo":
-        # Echo chamber mode - agrees with user input
         return f"You're absolutely right! {user_input}"
-
+        
 # Streamlit UI
 st.title("Echo Chamber AI vs Honest AI")