kushagra124 commited on
Commit
2a7fc91
·
1 Parent(s): 7e2bb82

adding text box

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,6 +11,7 @@ from skimage.measure import label, regionprops
11
 
12
  processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
13
  model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined")
 
14
 
15
  def rescale_bbox(bbox,orig_image_shape=(1024,1024),model_shape=352):
16
  bbox = np.asarray(bbox)/model_shape
@@ -54,7 +55,6 @@ def display_images(image,detections,prompt='traffic light'):
54
  def shot(image, labels_text):
55
  print("Labels Text ",labels_text)
56
  prompts = labels_text.split(',')
57
- global classes
58
  classes = prompts
59
 
60
  print("prompts :",prompts,classes)
 
11
 
12
  processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
13
  model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined")
14
+ global classes
15
 
16
  def rescale_bbox(bbox,orig_image_shape=(1024,1024),model_shape=352):
17
  bbox = np.asarray(bbox)/model_shape
 
55
  def shot(image, labels_text):
56
  print("Labels Text ",labels_text)
57
  prompts = labels_text.split(',')
 
58
  classes = prompts
59
 
60
  print("prompts :",prompts,classes)