Hugging Face's logo

Spaces:
PeterTP
/
Berries
Sleeping

App Files Community
PeterTP commited on
Commit
6e92d74
·
1 Parent(s): d6101c8

export.pkl

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. berry_model.pkl → export.pkl +0 -0
app.py CHANGED
@@ -5,9 +5,9 @@ from time import sleep
5
  import timm
6
  import gradio as gr
7
 
8
- learn = load_learner('berry_model.pkl')
9
 
10
- #catergories = ("strawberry", "blueberry", "raspberry", "cranberry", "mulberry", "other")
11
  categories = learn.dls.vocab
12
 
13
  def classify_image(img):
 
5
  import timm
6
  import gradio as gr
7
 
8
+ learn = load_learner('export.pkl')
9
 
10
+ #categories = ("strawberry", "blueberry", "raspberry", "cranberry", "mulberry", "other")
11
  categories = learn.dls.vocab
12
 
13
  def classify_image(img):
berry_model.pkl → export.pkl RENAMED
File without changes