Update app.py

app.py

@@ -110,16 +110,19 @@ convo = model.start_chat(history=[
   },
 ])
 
-
+#response = convo.send_message(
+#       messages=messages)
+#    return response.last.text
 
 def generate(messages):
-    response = convo.send_message(
-        messages=messages)
-    return response.last.text
-    
+    model = genai.GenerativeModel('gemini-pro')
+    response = model.generate_content(
+        messages,
+        stream=True,
+        generation_config=generation_config)   
 
 gr.ChatInterface(
-    fn=generate(messages),
+    fn=generate,
     chatbot=gr.Chatbot(show_label=False, avatar_images=(None, 'palm-logo.png'), show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
     title="PaLM-2",
     description="This is unofficial demo of ```PaLM-2``` based on ```Google API```. ```History/context``` memory does not work in this demo.",