obichimav commited on
Commit
f8358fa
·
verified ·
1 Parent(s): 09327f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -85,12 +85,10 @@ 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
- 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:
 
85
  return queries
86
  else: # DINO
87
  # DINO query format
88
+ query = f"a {words[:]}."
89
+ print("DINO query:", query)
90
+ return query
91
+
 
 
92
 
93
  def detect_objects(query_text):
94
  if state.current_image is None: