added back the yields

app.py

@@ -133,25 +133,25 @@ def bot_comms(input_text: str,
 
     if input_text == "mode":
         if llm_mode == "":
-            return "The mode is currently at Loki Default mode"
+            yield "The mode is currently at Loki Default mode"
         else:
-            return f"The current mode: {llm_mode}"
+            yield f"The current mode: {llm_mode}"
 
     if input_text == "check cuda":
         cuda_info = check_cuda()
-        return cuda_info
+        yield cuda_info
     
     if input_text == "switch to llama":
         llm_mode = input_text
-        return "Got it! Llama is now activate for your questions only 🦙"
+        yield "Got it! Llama is now activate for your questions only 🦙"
 
     if input_text == "switch to gpt-4o":
         llm_mode = input_text
-        return "Understood! GPT-4o is now hearing your responses only 👾"
+        yield "Understood! GPT-4o is now hearing your responses only 👾"
     
     if input_text == "switch to gpt-3.5-turbo":
         llm_mode = input_text
-        return "Done. GPT-3.5-turbo is ready for your questions! 🏃"
+        yield "Done. GPT-3.5-turbo is ready for your questions! 🏃"
 
     if llm_mode == "switch to llama":
         streamer = llama_generation(input_text=input_text, history=history, temperature=temperature, max_new_tokens=max_new_tokens)