Sunil1821/accident_car_detector

1from fastai.vision.all import *2import gradio as gr3import pickle 4 5categories = ('New Car', 'Car in an accident')6 7def greet(name):8    return "Hello" + name + "!!"9 10def is_car_in_accident(x):11    return x[0].isupper()12 13def classify_image(img):14    with open('model.pkl', 'r') as fp:15        learn = pickle.load(fp)16    pred, idx, probs = learn.predict(img)17    return dict(zip(categories, map(float, probs)))18# image = gr.inputs.Image(shape=(192, 192))19label = gr.Label()20examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']21iface = gr.Interface(fn=classify_image, inputs=gr.Image(), outputs=label)22iface.launch(inline=False)23