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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,11 +1,10 @@
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("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
  #)
 
1
  from fastai.vision.all import *
 
2
  from fastai.learner import load_learner
3
+ from huggingface_hub import from_pretrained_fastai, hf_hub_download
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
  #)