Add app.py and requirements.txt

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

app.py

@@ -0,0 +1,21 @@
+import requests
+from huggingface_hub import from_pretrained_fastai
+
+import gradio as gr
+
+repo_id = "artificeresearch/spiritvision"
+learner = from_pretrained_fastai(repo_id)
+
+
+def predict_fn(img):
+    """
+    :param img: img is a PIL image object
+    :return: prediction and probabilities
+    """
+    img = img.convert('RGB')
+    prediction, _, probs = learner.predict(img)
+    # print(f'{max(100 * probs):.2f}% {prediction} - {img}')
+    return f'{max(100 * probs):.2f}% {prediction} - {img}'
+
+
+gr.Interface(predict_fn, gr.inputs.Image(type='pil'), outputs='label').launch()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+timm