Update app.py

app.py

@@ -51,6 +51,19 @@ def gpt3_short(question,vqa_answer,caption):
     temperature=0.7,
     )
     answer = response.choices[0].text.strip()
+
+    llm_ans=answer
+    choice=set(["A","B","C","D"])
+    llm_ans=llm_ans.replace("\n"," ").replace(":"," ").replace("."," " ).replace(","," ")
+    sllm_ans=llm_ans.split(" ")
+    for cho in sllm_ans:
+      if cho in choice:
+         llm_ans=cho
+         break
+    if llm_ans not in choice:
+        llm_ans="A"
+    llm_ans=vqa_answer[ord(llm_ans)-ord("A")]
+    answer=llm_ans
     
     return answer
 def gpt3(question,vqa_answer,caption):