update app.py

app.py

@@ -1,5 +1,6 @@
-import gradio as gr
 import os
+import torch
+import gradio as gr
 from transformers import VisionEncoderDecoderModel, AutoFeatureExtractor, AutoTokenizer
 
 
@@ -18,10 +19,14 @@ def create_caption_transformer(img):
     return caption_text
 
 
+IMAGES_EXAMPLES_FOLDER = 'examples/'
+images = os.listdir(IMAGES_EXAMPLES_FOLDER)
+IMAGES_EXAMPLES = [IMAGES_EXAMPLES_FOLDER + img for img in images]
 model = VisionEncoderDecoderModel.from_pretrained(os.getcwd()).to('cpu')
 feature_extractor = AutoFeatureExtractor.from_pretrained('google/vit-base-patch16-224-in21k')
 tokenizer = AutoTokenizer.from_pretrained('gpt2')
 iface = gr.Interface(fn=create_caption_transformer,
                      inputs="image",
                      outputs='text',
+                     examples=IMAGES_EXAMPLES
                      ).launch(share=True)