adding text box

app.py

@@ -41,7 +41,7 @@ def detect_using_clip(image,prompts=[],threshould=0.4):
 
     return model_detections
 
-def display_images(image,detections,prompt):
+def visualize_images(image,detections,prompt):
     H,W = image.shape[:2]
     image_copy = image.copy()
     if prompt not in detections.keys():
@@ -63,10 +63,10 @@ def shot(image, labels_text,selected_categoty):
     model_detections  = detect_using_clip(image,prompts=prompts)
     print("detections :",model_detections)
     print("Ctegory ",selected_categoty)
-    return display_images(image=image,detections=model_detections,prompt=selected_categoty)
+    return visualize_images(image=image,detections=model_detections,prompt=selected_categoty)
 
 iface = gr.Interface(fn=shot,
-                    inputs = ["image","text",gr.Dropdown(classes, label="Category Label",info='Select Categories')],
+                    inputs = ["image","text","text"],
                     outputs="image",
                     description="Add a picture and a list of labels separated by commas",
                     title="Zero-shot Image Classification with Prompt ",