kushagra124 commited on
Commit
630b911
1 Parent(s): 739883b

adding text box

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,