Update app.py

app.py

@@ -59,7 +59,7 @@ def infer(img, text_queries, score_threshold, model):
   return result_labels
 
 # def query_image(img, text_queries, owl_threshold, dino_threshold):
-def query_image(img, text_queries, owl_threshold):
+def query_image(img, text_queries, owl_threshold, flag_output_korean):
     text_queries = text_queries
     text_queries = text_queries.split(",")
     owl_output = infer(img, text_queries, owl_threshold, "owl")
@@ -80,7 +80,12 @@ 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", value=english_candidate_labels_string), owl_threshold],
+    inputs=[
+        gr.Image(label="Input Image"), 
+        gr.Textbox(label="Candidate Labels", value=english_candidate_labels_string), 
+        owl_threshold,
+        gr.Checkbox(label="Output labels Korean")
+    ],
     # outputs=[owl_output, dino_output],
     outputs=[owl_output],
     title="OWLv2 Demo",