Update chatbot.py

chatbot.py

@@ -301,28 +301,28 @@ def model_inference( user_prompt, chat_history):
                             output += response.token.text
                             yield output
             
-            elif json_data["name"] == "image_generation":
-                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}")
-                gr.Info("Using Stability diffusion 3")
-                yield "Using Stability diffusion 3"
-                yield gr.Image(image)
+            # elif json_data["name"] == "image_generation":
+            #     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}")
+            #     gr.Info("Using Stability diffusion 3")
+            #     yield "Using Stability diffusion 3"
+            #     yield gr.Image(image)
                     
 
-            elif json_data["name"] == "video_generation":
-                query = json_data["arguments"]["query"]
-                gr.Info("Generating Video, Please wait 15 sec...")
-                yield "Generating Video, Please wait 15 sec..."
-                video = video_gen(f"{str(query)}")
-                yield gr.Video(video)
+            # elif json_data["name"] == "video_generation":
+            #     query = json_data["arguments"]["query"]
+            #     gr.Info("Generating Video, Please wait 15 sec...")
+            #     yield "Generating Video, Please wait 15 sec..."
+            #     video = video_gen(f"{str(query)}")
+                # yield gr.Video(video)
                 
             elif json_data["name"] == "image_qna":
                 inputs = llava(user_prompt, chat_history)