Hugging Face's logo

Spaces:
muddokon
/
PetClassifier
Sleeping

App Files Community
muddokon commited on
Commit
751cccd
·
verified ·
1 Parent(s): d30a0f2

Update app.py

Browse files

Added local pickle model file

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ 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
  #)
 
4
  import gradio as gr
5
  import skimage
6
 
7
+ learn = load_learner('PetNet50.pkl')
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
  #)