devdata commited on
Commit
ef33cde
·
1 Parent(s): 7e47f77

Update app.py

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