Upload 2 files

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import os
-from huggingface_hub import from_pretrained_fastai
+from transformers import pipeline
 
-pipe = from_pretrained_fastai("arnaucas/wildfire-classifier")
+pipe = pipeline("image-classification", model="arnaucas/wildfire-classifier")
 
 example_imgs = os.listdir("./examples")