Update app.py

app.py

@@ -103,7 +103,7 @@ def respond(message, history: list[tuple[str, str]], model, system_message, max_
         response += token
         yield response
 
-if mode == "text-to-text":
+while mode == "text-to-text":
     demo = gr.ChatInterface(
         respond,
         stop_btn = "Stop generation",
@@ -120,7 +120,7 @@ if mode == "text-to-text":
         ]
     )
 
-elif mode == "image-to-text":
+while mode == "image-to-text":
     demo = gr.ChatInterface(
         respond,
         stop_btn = "Stop generation",