Initial with fastai

exported_fastai.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2efcd6c1dcde8806b674d596d0a3dbf6b38d5eab048a5e8d9374dd53e69410eb
+size 22805785

requirements.txt

@@ -0,0 +1,7 @@
+gradio=="3.18.0"
+fastai=="2.7.10"
+fastcore=="1.5.28"
+pillow=="9.4.0"
+timm=="0.6.12"
+torch=="1.13.1"
+torchvision=="0.14.1"

src/app.py

@@ -0,0 +1,19 @@
+import gradio as gr
+from fastai.learner import load_learner
+from fastai.vision.core import PILImage
+
+# Load the model
+learn = load_learner('../exported_fastai.pkl', cpu=False)
+
+def classify_image(inp):
+    inp = PILImage.create(inp)
+    pred, _, _ = learn.predict(inp)
+    return pred
+
+iface = gr.Interface(
+    fn=classify_image,
+    inputs=gr.inputs.Image(),
+    outputs="text",
+    title="Fastai Classifier",
+    description="An example of using Fastai in Gradio.",
+).launch()