Update app.py

app.py

@@ -1,10 +1,11 @@
-from transformers import AutoFeatureExtractor, YolosForObjectDetection
-import gradio as gr
-from PIL import Image
-import torch
-import matplotlib.pyplot as plt
 import io
+import torch
 import numpy as np
+import gradio as gr
+import matplotlib.pyplot as plt
+
+from transformers import AutoFeatureExtractor, YolosForObjectDetection
+from PIL import Image
 
 
 COLORS = [
@@ -88,6 +89,7 @@ image_out = gr.components.Image()
 prob_threshold_slider = gr.components.Slider(
     minimum=0, maximum=1.0, step=0.01, value=0.7, label="Probability Threshold"
 )
+
 classes_to_show = gr.components.Textbox(
     placeholder="e.g. car, dog",
     label="Classes to filter (leave empty to detect all classes)",
@@ -97,5 +99,5 @@ gr.Interface(fn=model_inference,
              inputs=[image_in, prob_threshold_slider, classes_to_show],
              outputs=image_out,
              title=title,
-             examples=["CTH.png", "carplane.webp"],
+             examples=["00_plane.jpg", "01_car.jpg"],
              description=description).launch()