Update app.py

app.py

@@ -4,18 +4,10 @@ from transformers import pipeline
 generator = pipeline("text-generation", model="microsoft/DialoGPT-small")
 
 def chat_fn(message, history=None):
-    if history is None:
-        history = []
-    conversation = ""
-    for user_msg, bot_msg in history:
-        conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
-    conversation += f"User: {message}\nBot:"
     try:
-        response = generator(conversation, max_length=200, pad_token_id=50256)
-        bot_reply = response[0]['generated_text'].split("Bot:")[-1].strip()
+        response = generator(message, max_length=100, pad_token_id=50256)
+        bot_reply = response[0]['generated_text'].strip()
+        return bot_reply, history or []
     except Exception as e:
-        bot_reply = f"Error: {e}"
-    history.append((message, bot_reply))
-    return bot_reply, history
-
+        return f"Error: {e}", history or []
 gr.ChatInterface(fn=chat_fn).launch()