xxx1 commited on
Commit
cb03c06
·
1 Parent(s): 961aece

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def gpt3(question,vqa_answer,caption):
50
 
51
  def vle(input_image,input_text):
52
  vqa_answers = vqa_pipeline({"image":input_image, "question":input_text}, top_k=4)
53
- return [vqa.keys() for vqa in vqa_answers]
54
  def inference_chat(input_image,input_text):
55
  cap=caption(input_image)
56
  # inputs = processor(images=input_image, text=input_text,return_tensors="pt")
 
50
 
51
  def vle(input_image,input_text):
52
  vqa_answers = vqa_pipeline({"image":input_image, "question":input_text}, top_k=4)
53
+ return [" ".join([key for key,_ in vqa.items()] )for vqa in vqa_answers]
54
  def inference_chat(input_image,input_text):
55
  cap=caption(input_image)
56
  # inputs = processor(images=input_image, text=input_text,return_tensors="pt")