Hugging Face's logo

Spaces:
KathrynMercer
/
CatPatternClassifier
Sleeping

App Files Community
KathrynMercer commited on
Commit
e4eecf5
·
verified ·
1 Parent(s): 51d173f

Update app.py

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