Update app.py

app.py

@@ -28,25 +28,11 @@ def predict(image, max_length=64, num_beams=4):
 
 input = gr.inputs.Image(label="Upload any Image", type = 'pil', optional=True)
 output = gr.outputs.Textbox(type="text",label="Captions")
-examples = ["example1.jpg"]
-print("------------------------- 6 -------------------------\n")
-title = "Image to Text ViT with LORA"
-
-# interface = gr.Interface(
-            
-#         fn=predict,
-#         description=description,
-#         inputs = input,
-#         theme="grass",
-#         outputs=output,
-#         examples=examples,
-#         title=title,
-#     )
-# interface.launch(debug=True)
+examples_folder = os.path.join(os.path.dirname(__file__), "examples")
+examples = [os.path.join(examples_folder, file) for file in os.listdir(examples_folder)]
 
 with gr.Blocks() as demo:
     
-
     gr.HTML(
         """
         <div style="text-align: center; max-width: 1200px; margin: 20px auto;">
@@ -63,6 +49,7 @@ with gr.Blocks() as demo:
         
         </div>
         """)
+    
     with gr.Row():
             with gr.Column(scale=1):
                 img = gr.inputs.Image(label="Upload any Image", type = 'pil', optional=True)
@@ -71,8 +58,10 @@ with gr.Blocks() as demo:
                 out = gr.outputs.Textbox(type="text",label="Captions")   
                 
     button.click(predict, inputs=[img], outputs=[out])
+
+    
     gr.Examples(
-        examples=[os.path.join(os.path.dirname(__file__), "example1.jpg")],
+        examples=examples,
         inputs=img,
         outputs=out,
         fn=predict,