Update app.py

app.py

@@ -1,10 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-pipeline = pipeline(task="image-classification", model="MasleK/snails_snakes_slugs")
+from huggingface_hub import from_pretrained_fastai
+
+# repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
+repo_id = "MasleK/snails_snakes_slugs"
+
+learner = from_pretrained_fastai(repo_id)
+
+
 
 def predict(image):
-    predictions = pipeline(image)
+    predictions = learner.(image)
     return {p["label"]: p["score"] for p in predictions}
 
 gr.Interface(