torch360/app

1#!/usr/bin/env python2# coding: utf-83 4# ## Car Classifier5# Choose a PNG file of a Lamborghini, Mclaren, Pagani, Ferrari or a Koenigsegg to classify.6 7# In[1]:8 9 10from fastai.vision.all import *11from fastai.vision.widgets import *12import gradio as gr13 14 15# In[2]:16 17 18learn = load_learner('export.pkl')19 20 21# In[3]:22 23 24import warnings25warnings.filterwarnings("ignore")26 27 28# In[4]:29 30 31car_types = ('Ferrari','Koenigsegg', 'Lamborghini', 'Mclaren', 'Pagani')32 33 34# In[5]:35 36 37def image_classifier(im):38    pred = learn.predict(im)39    print(pred)40    return {car_types[i]: pred[2][i].item() for i in range(0,5)}41 42 43# In[6]:44 45 46iface = gr.Interface(47    image_classifier,48    gr.inputs.Image(shape=(224, 224)),49    gr.outputs.Label(num_top_classes=5),50    interpretation="default")51 52 53# In[7]:54 55 56iface.launch(share=True)57 58