Update app.py

app.py

@@ -1,5 +1,6 @@
 import io
 import torch
+import pathlib
 import numpy as np
 import gradio as gr
 import matplotlib.pyplot as plt
@@ -32,7 +33,7 @@ def model_inference(img, prob_threshold, classes_to_show):
 
     img = Image.fromarray(img)
 
-    pixel_values = feature_extractor(img, return_tensors="pil").pixel_values
+    pixel_values = feature_extractor(img, return_tensors="pt").pixel_values
 
     with torch.no_grad():
         outputs = model(pixel_values, output_attentions=True)
@@ -97,12 +98,13 @@ classes_to_show = gr.components.Textbox(
 )
 
 inputs = [image_in, prob_threshold_slider, classes_to_show]
-
+img_input = gr.Image(type='pil')
+example_images = gr.Dataset(components=[img_input], samples=[[path.as_posix()] for path in sorted(pathlib.Path('examples').rglob('*.*'))])
 
 gr.Interface(fn=model_inference,
              inputs=inputs,
              outputs=image_out,
              title=title,
              description=description,
-             examples=["carplane.webp", "CTH.png"]
+             examples=example_images
 ).launch()