Update app.py

app.py

@@ -12,7 +12,6 @@ images="dog.jpg"
 
 @spaces.GPU  
 def shot(input, labels_text):
-    # Check if the input is a URL or an uploaded image
     if isinstance(input, str) and (input.startswith("http://") or input.startswith("https://")):
         # Input is a URL
         response = requests.get(input)
@@ -21,25 +20,20 @@ def shot(input, labels_text):
         # Input is an uploaded image
         PIL_image = Image.fromarray(np.uint8(input)).convert('RGB')
     
-    # Split labels into a list
     labels = labels_text.split(",")
-    
-    # Perform the zero-shot image classification
     res = pipe(images=PIL_image, 
                candidate_labels=labels,
                hypothesis_template="This is a photo of a {}")
-    
-    # Return the classification results as a dictionary
     return {dic["label"]: dic["score"] for dic in res}
 
-# Define the Gradio interface
+# Define the Gradio interface with the updated components
 iface = gr.Interface(
     fn=shot, 
     inputs=[
-        gr.inputs.Textbox(label="Image URL (starting with http/https) or Upload Image"), 
-        "text"
+        gr.Textbox(label="Image URL (starting with http/https) or Upload Image"), 
+        gr.Textbox(label="Labels (comma-separated)")
     ], 
-    outputs="label", 
+    outputs=gr.Label(), 
     examples=[
         ["https://example.com/dog.jpg", "dog,cat,bird"], 
         ["https://example.com/germany.jpg", "germany,belgium,colombia"],
@@ -50,4 +44,4 @@ iface = gr.Interface(
 )
 
 # Launch the interface
-iface.launch()
+iface.launch()