Update app.py

app.py

@@ -16,7 +16,7 @@ def correct_words(text):
 def generate_response(message):
     global history
     for _ in range(2):  # Assuming you want two responses, adjust as needed
-        response = llm.generate(message)
+        response = llm.generate(message, max_length=50, num_return_sequences=1)[0]['generated_text']
         time.sleep(2)
         corrected_response = correct_words(response)
         history.append(corrected_response)