File size: 636 Bytes
666fd3d 937fe5b 666fd3d 7d6d4fb 666fd3d 7d6d4fb 666fd3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
import gradio as gr
from PIL import Image
from fastai.vision.all import *
def predict_image(image):
learner = load_learner('export.pkl')
img = PILImage.create(image)
pred = learner.predict(img)
return pred[0]
def create_interface():
image_input = gr.Image()
output = gr.Text()
iface = gr.Interface(
fn=predict_image,
inputs=image_input,
outputs=output,
title="Cat or Dog Classifier",
description="Upload an image to classify whether it's a cat or a dog."
)
return iface
if __name__ == "__main__":
iface = create_interface()
iface.launch()
|