nelbarman053's picture
version updated
7ce064d
raw
history blame contribute delete
708 Bytes
from fastai.vision.all import *
import gradio as gr
# import os
# os.system("pip uninstall -y gradio")
# os.system("pip install gradio==3.16.0")
# import pathlib
# temp = pathlib.PosixPath
# pathlib.PosixPath = pathlib.WindowsPath
model = load_learner('models/Recognizer.pkl')
labels = ['Lion', 'Tiger']
def recognize_image(image):
pred, id, probs = model.predict(image)
return dict(zip(labels, map(float, probs)))
image = gr.Image()
output_labels = gr.Label()
examples = [
'examples/unknown_1.jpg',
'examples/unknown_2.jpg'
]
iface = gr.Interface(fn=recognize_image, inputs = image, outputs = output_labels, examples = examples)
# iface.queue(api_open=True)
iface.launch(inline=False)