robinsk8a commited on
Commit
db6920d
·
1 Parent(s): 6733542

Updated with a huggingface model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,7 +1,9 @@
1
- import gradio as gr
2
  from fastai.vision.all import *
 
 
 
 
3
 
4
- learn = load_learner('sport_car_classifier.pkl')
5
  labels = learn.dls.vocab
6
  def predict(img):
7
  img = PILImage.create(img)
 
 
1
  from fastai.vision.all import *
2
+ from huggingface_hub import push_to_hub_fastai, from_pretrained_fastai
3
+ import gradio as gr
4
+
5
+ learn = from_pretrained_fastai("robinsk8a/exotic_cars_classifier")
6
 
 
7
  labels = learn.dls.vocab
8
  def predict(img):
9
  img = PILImage.create(img)