Spaces:
Sleeping
Sleeping
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,
|