Update chatbot.py

chatbot.py

@@ -212,7 +212,7 @@ client_mixtral = InferenceClient("NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO")
 client_llama = InferenceClient("meta-llama/Meta-Llama-3-8B-Instruct")
 client_mistral_nemo = InferenceClient("mistralai/Mistral-Nemo-Instruct-2407")
 
-@spaces.CPU(duration=60, queue=False)
+# @spaces.CPU(duration=60, queue=False)
 def model_inference( user_prompt, chat_history):
     if user_prompt["files"]:
         inputs = llava(user_prompt, chat_history)
@@ -305,14 +305,15 @@ def model_inference( user_prompt, chat_history):
                 query = json_data["arguments"]["query"]
                 gr.Info("Generating Image, Please wait 10 sec...")
                 yield "Generating Image, Please wait 10 sec..."
-                try:
-                    image = image_gen(f"{str(query)}")
-                    yield gr.Image(image[1])
-                except:
-                    client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
-                    seed = random.randint(0,999999)
-                    image = client_sd3.text_to_image(query, negative_prompt=f"{seed}")
-                    yield gr.Image(image)
+                # try:
+                #     image = image_gen(f"{str(query)}")
+                #     yield gr.Image(image[1])
+                # except:
+                client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
+                seed = random.randint(0,999999)
+                image = client_sd3.text_to_image(query, negative_prompt=f"{seed}")
+                gr.Info("Using Stability diffusion 3")
+                yield gr.Image(image)
                     
 
             elif json_data["name"] == "video_generation":