Update app.py
Browse files
app.py
CHANGED
@@ -85,9 +85,12 @@ def format_query_for_model(text_input, model_type="owlv2"):
|
|
85 |
return queries
|
86 |
else: # DINO
|
87 |
# DINO query format
|
88 |
-
query = f"a {words[0]}."
|
89 |
-
print("DINO query:", query)
|
90 |
-
return query
|
|
|
|
|
|
|
91 |
|
92 |
def detect_objects(query_text):
|
93 |
if state.current_image is None:
|
|
|
85 |
return queries
|
86 |
else: # DINO
|
87 |
# DINO query format
|
88 |
+
# query = f"a {words[0]}."
|
89 |
+
# print("DINO query:", query)
|
90 |
+
# return query
|
91 |
+
queries = ["a photo of " + obj for obj in words]
|
92 |
+
print("DINO query:", queries)
|
93 |
+
return queries
|
94 |
|
95 |
def detect_objects(query_text):
|
96 |
if state.current_image is None:
|