ramonvictorn's picture
feat: addmin better examples and remove old cod
f96a346
raw
history blame contribute delete
No virus
596 Bytes
import gradio as gr
from fastai.vision.all import *
from huggingface_hub import from_pretrained_fastai
learn = from_pretrained_fastai("ramonvictorn/brazilian_car_check")
categories = ('Imported', 'Nacional')
def classify_image(img):
prd, idx, props = learn.predict(img)
return dict(zip(categories, map(float, props)))
image = gr.Image(shape=(192, 192), type="pil", label="Upload a car candidate")
label = gr.outputs.Label()
intf = gr.Interface(
fn=classify_image,
inputs=image,
outputs=label, examples=["ferrari.jpg","kwid.png", "celta.jpeg"]
)
intf.launch(inline=True)