muddokon commited on
Commit
ca2787b
·
verified ·
1 Parent(s): bbb9444

Update app.py

Browse files

switched to different model due to issue with mlp module in timm

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from fastai.learner import load_learner
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
  #)
 
4
  import gradio as gr
5
  import skimage
6
 
7
+ learn = from_pretrained_fastai("espejelomar/fastai-pet-breeds-classification")
8
+ #learn = from_pretrained_fastai("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k")
9
  #learn = load_learner(
10
  # hf_hub_download("kurianbenoy/course_v5_lesson2_pets_convnext_base_in22k", "model.pkl")
11
  #)