adding text box

app.py

@@ -52,20 +52,23 @@ def display_images(image,detections,prompt='traffic light'):
 
 
 def shot(image, labels_text):
-    print(labels_text)
+    print("Labels Text ",labels_text)
     prompts = labels_text.split(',')
     global classes 
     classes = prompts
 
+    print("prompts :",prompts,classes)
+    print("Image shape ",image.shape )
+
     detections  = detect_using_clip(image,prompts=prompts)
-    
+    print("detections :",detections)
     return detections
 
 def add_text(text):
     labels = text.split(',')
     return labels 
 
-inputt = gr.Image(type="numpy", label="Input Image for Classification")
+# inputt = gr.Image(type="numpy", label="Input Image for Classification")
 
 # with gr.Blocks(title="Zero Shot Object ddetection using Text Prompts") as demo :
 #     gr.Markdown(
@@ -94,9 +97,11 @@ inputt = gr.Image(type="numpy", label="Input Image for Classification")
 
 # demo.launch()
 iface = gr.Interface(fn=shot,
-                    inputs = ["image","text"],
+                    inputs = ["image","text",gr.Dropdown(classes, label="Category Label",info='Select Categories')],
                     outputs="label",
-                    # examples=[["images/room.jpg","bed,table,plant"]],
+                    description="Add a picture and a list of labels separated by commas",
+                    title="Zero-shot Image Classification with Prompt ",
+                    examples=[["images/room.jpg","bed,table,plant"]],
                     # allow_flagging=False, 
                     # analytics_enabled=False,
                 )