Update app.py

app.py

@@ -15,7 +15,7 @@ generation_config = {
 }
 
 model = genai.GenerativeModel(
-  model_name="gemini-1.5-pro",
+  model_name="gemini-1.5-flash",
   generation_config=generation_config,
   # safety_settings = Adjust safety settings
   # See https://ai.google.dev/gemini-api/docs/safety-settings
@@ -71,17 +71,19 @@ AI=create_chat_session()
 def   get_answer_ai(text):
       global AI
       try:
-          response = AI.send_message(text)
-          return response.text
+          response = AI.send_message(text,stream=True)
+          return response
 
           
       except :
           AI=create_chat_session()
-          response = AI.send_message(text)
-          return response.text
+          response = AI.send_message(text,stream=True)
+          return response
 def dash(text):
-    textai=get_answer_ai(text)
-    return textai
+    response=get_answer_ai(text)
+    for chunk in  response:
+        yield chunk.text
+    # return textai
 
 demo = gr.Interface(fn=dash, inputs="text", outputs="text")
 demo.launch()