Update app.py

app.py

@@ -15,14 +15,16 @@ def correct_words(text):
 
 def generate_response(message):
     global history
-    for text in llm.generate(message):
-        response = ''.join(text)
-        time.sleep(0.1)
+    for _ in range(2):  # Assuming you want two responses, adjust as needed
+        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)
         yield ' '.join(history)
     # Clear the history list after the last response
     history = ["Chatbot:"]
+    # Clear the history list after the last response
+    history = ["Chatbot:"]
 
 def chatbot(message, history):
     response_generator = generate_response(message)