Spaces:
Running
Running
kushagra124
commited on
Commit
·
630b911
1
Parent(s):
739883b
adding text box
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def display_images(image,detections,prompt='traffic light'):
|
|
52 |
return image_copy
|
53 |
|
54 |
|
55 |
-
def shot(image, labels_text):
|
56 |
print("Labels Text ",labels_text)
|
57 |
prompts = labels_text.split(',')
|
58 |
classes = prompts
|
@@ -62,7 +62,7 @@ def shot(image, labels_text):
|
|
62 |
|
63 |
detections = detect_using_clip(image,prompts=prompts)
|
64 |
print("detections :",detections)
|
65 |
-
|
66 |
return 0
|
67 |
|
68 |
iface = gr.Interface(fn=shot,
|
|
|
52 |
return image_copy
|
53 |
|
54 |
|
55 |
+
def shot(image, labels_text,selected_categoty):
|
56 |
print("Labels Text ",labels_text)
|
57 |
prompts = labels_text.split(',')
|
58 |
classes = prompts
|
|
|
62 |
|
63 |
detections = detect_using_clip(image,prompts=prompts)
|
64 |
print("detections :",detections)
|
65 |
+
print("Ctegory ",selected_categoty)
|
66 |
return 0
|
67 |
|
68 |
iface = gr.Interface(fn=shot,
|