Mohammed Albrayh
Update app.py
d8c399e
raw
history blame contribute delete
622 Bytes
import gradio as gr
from fastai.vision.all import *
#Windows fix
import pathlib
plt = platform.system()
if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
learn = load_learner('SaudiLandmarkClassifier.pkl')
def predict(input):
# image = gr.Image(height=224,type="pil",image_obj=input)
image = PILImage.create(input).resize((224,224))
results = learn.predict(image)[0]
return results
iface = gr.Interface(fn=predict, inputs="image", outputs="text", examples=['Examples/A.png', 'Examples/B.png', 'Examples/C.png'])
iface.launch()