Upload app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 from transformers import pipeline
 from pathlib import Path
+from PIL import Image
+import numpy as np
 
 example_imgs = ["examples/img0.jpg", 
                 "examples/img1.jpg",
@@ -11,8 +13,7 @@ example_imgs = ["examples/img0.jpg",
 pipe = pipeline("image-classification", model="arnaucas/wildfire-classifier")
 
 def inference(image):
-    if image in example_imgs:
-        image = "./examples/" + image
+    image = Image.fromarray(np.uint8(image)).convert('RGB')
     output = pipe(image)
     result = {item['label']: item['score'] for item in output}
     return result