Update app.py

app.py

@@ -36,7 +36,7 @@ def predict_image(img, conf_threshold, iou_threshold):
 
     return im
 
-current_directory = "./image" 
+# current_directory = ./image 
 
 iface = gr.Interface(
     fn=predict_image,
@@ -49,8 +49,8 @@ iface = gr.Interface(
     title="Fire Detection using YOLOv8n on Gradio",
     description="Upload images for inference. The Ultralytics YOLOv8n trained model is used for inference.",
     examples=[
-        [os.path.join(current_directory, "fire_image_1.jpg")],
-        [os.path.join(current_directory, "fire_image_3.jpg")],
+        [os.path.join(os.path.dirname(__file__), "image/fire_image_1.jpg")],
+        [os.path.join(os.path.dirname(__file__), "image/fire_image_3.jpg")],
     ]
 )