andrewkatumba commited on
Commit
4550f69
1 Parent(s): 8c692eb

Fix color issues

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def infer(img, text_queries, score_threshold, model):
49
  return result_labels
50
 
51
  def query_image(img, text_queries, dino_threshold):
52
- text_queries = text_queries.split(",")
53
  dino_output = infer(img, text_queries, dino_threshold, "dino")
54
  annotations = []
55
  for box, label in dino_output:
 
49
  return result_labels
50
 
51
  def query_image(img, text_queries, dino_threshold):
52
+ text_queries = [query.strip() for query in text_queries.split(",")]
53
  dino_output = infer(img, text_queries, dino_threshold, "dino")
54
  annotations = []
55
  for box, label in dino_output: