merve HF staff commited on
Commit
6ec7588
1 Parent(s): fee2c8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -51,7 +51,10 @@ def infer(img, text_queries, score_threshold, model):
51
  continue
52
  if model == "owl":
53
  label = text_queries[label.cpu().item()]
54
- result_labels.append((box, label))
 
 
 
55
  return result_labels
56
 
57
  def query_image(img, text_queries, owl_threshold, dino_threshold):
 
51
  continue
52
  if model == "owl":
53
  label = text_queries[label.cpu().item()]
54
+ result_labels.append((box, label))
55
+ elif model == "dino":
56
+ if label != "":
57
+ result_labels.append((box, label))
58
  return result_labels
59
 
60
  def query_image(img, text_queries, owl_threshold, dino_threshold):