Gradio space

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+from fastai.learner import load_learner
+from fastai.vision.core import PILImage
+from huggingface_hub import hf_hub_download
+
+
+learner = load_learner(hf_hub_download("jdmartinev/intel_image_classification_fastai","model.pkl"))
+
+def classify_image(image):
+    img = PILImage.create(image)
+    pred_class, pred_idx, outputs = learn.predict(img)
+    return f"Predicted class: {pred_class}"
+
+interface = gr.Interface(fn=classify_image, 
+                         inputs=gr.inputs.Image(type="file", label="Upload an Image"), 
+                         outputs="text")
+
+interface.launch()

requeriments.txt

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