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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
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: