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='traffic light'):
+def display_images(image,detections,prompt):
     H,W = image.shape[:2]
     image_copy = image.copy()
     if prompt not in detections.keys():
@@ -60,10 +60,10 @@ def shot(image, labels_text,selected_categoty):
     print("prompts :",prompts,classes)
     print("Image shape ",image.shape )
 
-    detections  = detect_using_clip(image,prompts=prompts)
-    print("detections :",detections)
+    model_detections  = detect_using_clip(image,prompts=prompts)
+    print("detections :",model_detections)
     print("Ctegory ",selected_categoty)
-    return 0
+    return display_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')],