Spaces:
Sleeping
Sleeping
File size: 943 Bytes
4df8c69 f112b06 877dcac 4df8c69 | 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 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | from fastai.vision.all import load_learner
import gradio as gr
car_logo = [
"Audi car logo",
"BMW car logo",
"Chevrolet car logo",
"Dodge car logo",
"Ford car logo",
"Honda car logo",
"Hyundai car logo",
"Jaguar car logo",
"Jeep car logo",
"Kia car logo",
"Land Rover car logo",
"Lexus car logo",
"Mazda car logo",
"Mercedes-Benz car logo",
"Nissan car logo",
"Porsche car logo",
"Subaru car logo",
"Tesla car logo",
"Toyota car logo",
"Volkswagen car logo"]
model = load_learner("models/logo_recognizer_v1.pkl")
def recognize_image(image):
pred, idx, probs = model.predict(image)
return dict(zip(car_logo, map(float,probs)))
image = gr.inputs.Image()
label = gr.outputs.Label()
examples = [
'test_images/lexus-logo.png',
'test_images/Mazda-Logo.png',
'test_images/toyota-logo.png']
demo = gr.Interface(fn=recognize_image, inputs="image", outputs="label", examples = examples)
demo.launch(inline = False, share = True)
|