Update app.py

app.py

@@ -23,7 +23,7 @@ def caption(input_image):
     inputs["num_beams"] = 4
     inputs['num_return_sequences'] =4
     out = cap_model.generate(**inputs)
-    return "\n".join(processor.decode(out[0], skip_special_tokens=True))
+    return "\n".join(processor.decode(out, skip_special_tokens=True))
 import openai
 openai.api_key="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"
 def gpt3(question,vqa_answer,caption):
@@ -69,18 +69,18 @@ with gr.Blocks(
                     with gr.Row():
                         clear_button = gr.Button(value="Clear", interactive=True)
                         submit_button = gr.Button(
-                            value="Submit", interactive=True, variant="primary"
+                            value="Submit_VQA", interactive=True, variant="primary"
                         )
                     cap_submit_button = gr.Button(
-                            value="Submit", interactive=True, variant="primary"
+                            value="Submit_CAP", interactive=True, variant="primary"
                         )
                     gpt3_submit_button = gr.Button(
-                            value="Submit", interactive=True, variant="primary"
+                            value="Submit_GPT3", interactive=True, variant="primary"
                         )
         with gr.Column():
             caption_output = gr.Textbox(lines=0, label="VQA Output(模型答案输出)")
             caption_output_v1 = gr.Textbox(lines=0, label="Caption Output(模型caption输出)")
-            gpt3_output_v1 = gr.Textbox(lines=0, label="GPT3 Output(模型caption输出)")
+            gpt3_output_v1 = gr.Textbox(lines=0, label="GPT3 Output(GPT3输出)")
             
         image_input.change(
             lambda: ("", "", []),