Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-import threading
 from transformers import pipeline
 
 # Load the hotdog-not-hotdog model
@@ -8,42 +7,33 @@ hotdog_pipeline = pipeline(task="image-classification", model="julien-c/hotdog-n
 # Load the chatbot model (DialoGPT)
 chatbot_pipeline = pipeline(task="text-generation", model="microsoft/DialoGPT-medium")
 
-def hotdog_predict(input_img):
-    # Predict hotdog or not
-    hotdog_predictions = hotdog_pipeline(input_img)
-
-    return input_img, {
-        "Hotdog Classification": {p["label"]: p["score"] for p in hotdog_predictions},
-    }
-
-def chatbot_predict(chat_input):
-    # Generate chatbot response
-    chatbot_response = chatbot_pipeline(chat_input, max_length=50)[0]['generated_text']
-
-    return chat_input, {
-        "Chatbot Response": chatbot_response
-    }
-
-hotdog_interface = gr.Interface(
-    fn=hotdog_predict,
-    inputs=gr.Image(label="Select hot dog candidate", sources=['upload', 'webcam'], type="pil"),
+def predict(input_data, option):
+    if option == "Image Classification":
+        # Predict hotdog or not
+        hotdog_predictions = hotdog_pipeline(input_data)
+        return input_data, {
+            "Hotdog Classification": {p["label"]: p["score"] for p in hotdog_predictions},
+        }
+    elif option == "Chatbot":
+        # Generate chatbot response
+        chatbot_response = chatbot_pipeline(input_data, max_length=50)[0]['generated_text']
+        return input_data, {
+            "Chatbot Response": chatbot_response
+        }
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=[
+        gr.Image(label="Select hot dog candidate", sources=['upload', 'webcam'], type="pil"),
+        gr.Dropdown(label="Choose Option", choices=["Image Classification", "Chatbot"])
+    ],
     outputs=[
         gr.Image(label="Processed Image"),
         gr.Label(label="Hotdog Classification", num_top_classes=2),
+        gr.Textbox(label="Chatbot Response", type="text"),
     ],
-    title="Hot Dog? Or Not?",
-)
-
-chatbot_interface = gr.Interface(
-    fn=chatbot_predict,
-    inputs=gr.Textbox(label="Chatbot Input", placeholder="Type something to chat with the bot", type="text"),
-    outputs=gr.Textbox(label="Chatbot Response", type="text"),
-    title="Chat with the Bot",
+    title="Hot Dog? Or Chat with the Bot?",
 )
 
-def launch_interfaces():
-    hotdog_interface.launch()
-    chatbot_interface.launch()
-
 if __name__ == "__main__":
-    threading.Thread(target=launch_interfaces).start()
+    iface.launch()