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

Update app.py

Browse files

Back to pretrained fast ai method approach

Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,10 +4,10 @@ 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):
 
4
  import gradio as gr
5
  import skimage
6
 
7
+ learn = from_pretrained_fastai("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k")
8
+ #learn = load_learner(
9
+ # hf_hub_download("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k", "model.pkl")
10
+ #)
11
 
12
  labels = learn.dls.vocab
13
  def predict(img):