Update app.py

app.py

@@ -1,20 +1,24 @@
 import gradio as gr
 from transformers import pipeline
 
-pipeline_hotdog = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+# Load the hotdog-not-hotdog model
+hotdog_pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
+# Load the chatbot model (DialoGPT)
 chatbot_pipeline = pipeline(task="text-generation", model="microsoft/DialoGPT-medium")
 
-def predict(input_img,chat_input):
-    hotdog_predictions = pipeline_hotdog(input_img)
-        # Generate chatbot response
+def predict(input_img, chat_input):
+    # Predict hotdog or not
+    hotdog_predictions = hotdog_pipeline(input_img)
+
+    # Generate chatbot response
     chatbot_response = chatbot_pipeline(chat_input, max_length=50)[0]['generated_text']
 
     return input_img, {
         "Hotdog Classification": {p["label"]: p["score"] for p in hotdog_predictions},
         "Chatbot Response": chatbot_response
     }
-    
+
 gradio_app = gr.Interface(
     fn=predict,
     inputs=[