Update app.py

app.py

@@ -33,6 +33,7 @@ import torch
 import cv2
 import numpy as np
 import gradio as gr
+import os
 
 model = YOLOv3Lightning()
 model.load_state_dict(torch.load("yolov3_model_without_75_mosaic.pth", map_location=torch.device('cpu')), strict=False)
@@ -88,13 +89,12 @@ def process_image_and_plot(image, model, scaled_anchors):
 
     return fig,cam_image
 
+images_folder = "examples"
 
-examples = [
-    ["/images/automobile.jpg"],
-    ["/images/cycle.jpg"],
-    ["images/dog-kitten.jpg"],
-    ["images/human.jpg"],
-]
+examples = [[os.path.join(images_folder,"automobile.jpg")],
+            [os.path.join(images_folder,"cycle.jpg")],
+            [os.path.join(images_folder,"dog-kitten.jpg")],
+            [os.path.join(images_folder,"human.jpg")]]
 
 def processed_image(image):
     figure,gradcam = process_image_and_plot(image, model, scaled_anchors)