Upload 2 files

app.py

@@ -1,12 +1,18 @@
 import gradio as gr
 import os
 from transformers import pipeline
+from pathlib import Path
 
-pipe = pipeline("image-classification", model="arnaucas/wildfire-classifier")
+example_imgs = ["examples/img0.jpg", 
+                "examples/img1.jpg",
+                "examples/img2.jpg",
+                "examples/img3.jpg"]
 
-example_imgs = os.listdir("./examples")
+pipe = pipeline("image-classification", model="arnaucas/wildfire-classifier")
 
 def inference(image):
+    if image in example_imgs:
+        image = "./examples/" + image
     output = pipe(image)
     result = {item['label']: item['score'] for item in output}
     return result