Update app.py

app.py

@@ -19,8 +19,11 @@ image_transform = testing_image_transform()
 question_transform = ALBEFTextTransform(add_end_token=False)
 answer_transform = ALBEFTextTransform(do_pre_process=False)
 
+vqa_data = json.load(open(data_dir + "vqa_data.json", "r"))
 answer_list = json.load(open(data_dir + "answer_list.json", "r"))
 
+examples = [[data['image'], data['question']] for data in vqa_data]
+
 def infer(image, question):
   images = [image]
   image_input = [image_transform(image) for image in images]
@@ -49,12 +52,7 @@ demo = gr.Interface(
     fn=infer,
     inputs=[gr.Image(label='image', type='pil', image_mode='RGB'), gr.Text(label='question')],
     outputs=gr.Text(label='answer'),
-    # examples=[
-    #     ['vqav2.png', 'What sport is this?'],
-    #     ['vizwiz.jpeg', 'What piece of meat have I taken out of the freezer?'],
-    #     ['aqua.png', 'what does bol lean nonchalantly on'],
-    #     ['robotvqa.png', 'How many silver spoons are there?'],
-    # ]
+    examples=examples
 )
 
 demo.launch()