Update app.py

add - initial input strings

app.py

@@ -69,6 +69,9 @@ def query_image(img, text_queries, owl_threshold):
     # return (img, owl_output), (img, dino_output)
     return (img, owl_output)
 
+english_candidate_labels = ["hat", "sunglass", "hair band", "glove", "arm sleeve", "watch", "singlet", "t-shirts", "energy gel", "half pants", "socks", "shoes"]
+korean_candidate_labels = ["모자", "썬글라스", "헤어밴드", "장갑", "팔토시", "시계", "싱글렛", "티셔츠", "에너지젤", "쇼츠바지", "양말", "신발"]
+english_candidate_labels_string = ",".join(english_candidate_labels)
 
 owl_threshold = gr.Slider(0, 1, value=0.16, label="OWL Threshold")
 # dino_threshold = gr.Slider(0, 1, value=0.12, label="Grounding DINO Threshold")
@@ -77,7 +80,7 @@ owl_output = gr.AnnotatedImage(label="OWL Output")
 demo = gr.Interface(
     query_image,
     # inputs=[gr.Image(label="Input Image"), gr.Textbox(label="Candidate Labels"), owl_threshold, dino_threshold],
-    inputs=[gr.Image(label="Input Image"), gr.Textbox(label="Candidate Labels"), owl_threshold],
+    inputs=[gr.Image(label="Input Image"), gr.Textbox(label=english_candidate_labels_string), owl_threshold],
     # outputs=[owl_output, dino_output],
     outputs=[owl_output],
     title="OWLv2 Demo",