call secret IAW docs

app.py

@@ -2,11 +2,12 @@ import os
 from huggingface_hub import from_pretrained_fastai
 
 import gradio as gr
+from gradio.components import Image
 
+HF_TOKEN = os.getenv("HF_TOKEN")
 repo_id = "artificeresearch/spiritvision"
-learner = from_pretrained_fastai(repo_id)
+learner = from_pretrained_fastai(repo_id, token=HF_TOKEN)
 labels = learner.dls.vocab
-HF_TOKEN = os.environ.get("HF_TOKEN")
 
 
 def predict_fn(img):
@@ -14,12 +15,11 @@ def predict_fn(img):
     :param img: img is a PIL image object
     :return: prediction and probabilities
     """
-    img = img.convert('RGB')
-    # print(f'{max(100 * probs):.2f}% {prediction} - {img}')
     pred, pred_idx, probs = learner.predict(img)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 
 gr.Interface(predict_fn,
-             gr.inputs.Image(type='pil', shape=(512, 512)),
-             outputs=gr.outputs.Label(num_top_classes=3), token=HF_TOKEN).launch()
+             Image(shape=(512, 512)),
+             outputs=gr.outputs.Label(num_top_classes=3),
+             token=HF_TOKEN).launch()