#/export 

import gradio as gr
from fastai.vision.all import *


def is_bike(x):
    return x[0].isupper()



learn = load_learner("bike.pkl")


categories =("road_bike","downhill_bike","bmx")

def classsify_image(img):
    pred , idx,probs = learn.predict(img)
    return dict(zip(categories,map(float,probs)))


image = gr.inputs.Image(shape=(192,192))
label = gr.outputs.Label()
examples = ["bike_test1.jfif","downhill1.jfif","road1.jfif"]

intf = gr.Interface(fn=classsify_image,inputs=image,outputs=label,examples=examples)
intf.launch(inline=False,share=True)