Update app.py

app.py

@@ -1,6 +1,10 @@
+ 
+import os
+
+examples = os.listdir() 
+
 import torch 
 import re 
-import os
 import gradio as gr
 from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel 
 
@@ -25,12 +29,12 @@ def predict(image,max_length=64, num_beams=4):
 
 input = gr.inputs.Image(label="Upload your Image", type = 'pil', optional=True)
 output = gr.outputs.Textbox(type="auto",label="Captions")
-#examples = [f"example{i}.jpg" for i in range(1,5)]
-examples = os.listdir()
+examples = [f"example{i}.jpg" for i in range(1,7)]
+
 description= "Image captioning application made using transformers"
 title = "Image Captioning 🖼️"
 
-article = "Created By : Xaheen "
+article = "Created By : Shreyas Dixit "
 
 interface = gr.Interface(
         fn=predict,