Upload app.py

app.py

@@ -43,8 +43,8 @@ def instruct_generate(
         temperature: A value controlling the randomness of the sampling process. Higher values result in more random
     """
     # scene_name = os.path.basename(img_path).split(".")[0]
-    # if input in input_value_list:
-    #     input = example_dict[scene_name]["input"]
+    if input in input_value_2_real.keys():
+        input = input_value_2_real[input]
     if "..." in input:
         input = input.replace("...", "")
     output = [prompt, input, max_new_tokens, top_k, temperature]
@@ -61,9 +61,10 @@ max_batch_size = 1
 with open(example_path, 'r') as f:
     content = f.read()
     example_dict = json.loads(content)
-input_value_list = []
+input_value_2_real = {}
 for scene_id, scene_dict in example_dict.items():
-    input_value_list.append(scene_dict["input_display"])
+    input_value_2_real[scene_dict["input_display"]] = scene_dict["input"]
+examples_img_list = glob.glob("caption_demo/*.png")
 
 def create_instruct_demo():
     with gr.Blocks() as instruct_demo:
@@ -90,7 +91,6 @@ def create_instruct_demo():
         inputs = [scene_img, instruction, object_list, max_len, top_k, temp]
 
         # 接下来设定具体的example格式
-        examples_img_list = glob.glob("caption_demo/*.png")
         examples = []
         for example_img_one in examples_img_list:
             scene_name = os.path.basename(example_img_one).split(".")[0]