Spaces:
Sleeping
Sleeping
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() |