muddokon commited on
Commit
f191270
·
verified ·
1 Parent(s): 88b33b1

Update app.py

Browse files

Updated model download method

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