Upload app.py

app.py

@@ -88,16 +88,17 @@ def create_instruct_demo():
             with gr.Column():
                 outputs = gr.Textbox(lines=20, label="Output")
 
-        inputs = [scene_img, instruction, object_list, max_len, temp, top_k]
+        # inputs = [scene_img, instruction, object_list, max_len, temp, top_k]
+        inputs = [scene_img, instruction, object_list]
 
         # 接下来设定具体的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]
-            example_object_list = example_dict[scene_name]["input_display"]
+            example_object_list = example_dict[scene_name]["input"]
             example_instruction = example_dict[scene_name]["instruction"]
-            example_one = [example_img_one, example_instruction, example_object_list, 1024, 0.8, 200]
+            example_one = [example_img_one, example_instruction, example_object_list]
             examples.append(example_one)
 
         gr.Examples(
@@ -107,9 +108,7 @@ def create_instruct_demo():
             fn=instruct_generate,
             cache_examples=os.getenv('SYSTEM') == 'spaces'
         )
-        for example_one in examples:
-            scene_name = os.path.basename(example_one[0]).split(".")[0]
-            example_one[2] = example_dict[scene_name]["input"]
+        inputs = inputs + [max_len, temp, top_k]
         run_botton.click(fn=instruct_generate, inputs=inputs, outputs=outputs)
     return instruct_demo