Update app.py

app.py

@@ -24,17 +24,20 @@ def caption(input_image):
     inputs['num_return_sequences'] =4
     out = model.generate(**inputs)
     return "\n".join(processor.decode(out[0], skip_special_tokens=True))
-def gpt3(input_text):
+import openai
+openai.api_key=openai_appkey
+def gpt3(question,vqa_answer,caption):
+    prompt=caption+"\n"+question+"\n"+vqa_answer+"\n Tell me the right answer."
     response = openai.Completion.create(
     engine="text-davinci-003",
-    prompt=input_text,
+    prompt=prompt,
     max_tokens=10,
     n=1,
     stop=None,
     temperature=0.7,
     )
     answer = response.choices[0].text.strip()
-    return answer
+    return "input_text:\n"+prompt+"\n\n output_answer:\n"+answer
 
     
 def inference_chat(input_image,input_text):
@@ -71,9 +74,13 @@ with gr.Blocks(
                     cap_submit_button = gr.Button(
                             value="Submit", interactive=True, variant="primary"
                         )
+                    gpt3_submit_button = gr.Button(
+                            value="Submit", 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输出)")
             
         image_input.change(
             lambda: ("", "", []),
@@ -111,6 +118,15 @@ with gr.Blocks(
                         ],
                         [caption_output_v1],
                     )
+        gpt3_submit_button.click(
+                        gpt3,
+                        [
+                            chat_input,
+                           caption_output ,
+                            caption_output_v1,
+                        ],
+                        [gpt3_output_v1],
+                    )
 
    # examples = gr.Examples(
    #     examples=examples,