File size: 639 Bytes
d156ff9 8c9d8e4 8622689 5200e36 8622689 5200e36 8622689 8e95bbe 5200e36 8622689 | 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 | import platform
import pathlib
plt = platform.system()
if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
from fastai.vision.all import*
import gradio as gr
from gradio import Label
def oldornot(x): return x[0].isupper()
#/export
learn = load_learner('model.pkl')
#/export
categories = ['New Car','Old Car']
def classify_image(img):
pred,idx,probs = learn.predict(img)
return dict(zip(categories,map(float,probs)))
examples = ['new_car_2.jpg','new_car_4.jpg','old_car_7.jpg','old_car_12.jpg']
intf = gr.Interface(fn=classify_image, inputs=gr.Image(), outputs=Label(), examples=examples)
intf.launch(inline=False)
|