Update app.py

app.py

@@ -4,15 +4,13 @@ import gradio as gr
 
 
 chatbot = pipeline("conversational", model="ericzhou/DialoGPT-Medium-Rick_v2")
-conversation = None # initialization
+
 def predict(sentence,history) : 
-  global conversation
-  if not conversation : 
-    # initial generation
-    conversation = Conversation(sentence)
-  else : 
-    # mid conversation
-    conversation.add_user_input(sentence)
+  conversation = Conversation()
+  for hist in history: 
+    conversation.add_user_input(hist[0])
+    conversation.append_response(hist[1])
+  conversation.add_user_input(sentence)
   conversation = chatbot(conversation)
   return conversation.generated_responses[-1]