Spaces:
Sleeping
Sleeping
Commit
·
4550f69
1
Parent(s):
8c692eb
Fix color issues
Browse files
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:
|