restore previous commit

app.py

@@ -1,23 +1,17 @@
 import os
-from huggingface_hub import hf_hub_download
-from fastai.learner import load_learner
+from huggingface_hub import from_pretrained_fastai
 
 import gradio as gr
 from gradio.components import Image
 from gradio import Label
 
-SECRET_TOKEN = os.environ["SECRET_TOKEN"]
-
-# learner = from_pretrained_fastai("artificeresearch/spiritvision")
-learner = load_learner(hf_hub_download("artificeresearch/spiritvision", "model.pkl", token=SECRET_TOKEN))
+SECRET_TOKEN = os.getenv("SECRET_TOKEN")
+repo_id = "artificeresearch/spiritvision"
+learner = from_pretrained_fastai(repo_id)
 labels = learner.dls.vocab
 
 
 def predict_fn(img):
-    """
-    :param img: img is a PIL image object
-    :return: prediction and probabilities
-    """
     pred, pred_idx, probs = learner.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}