Hugging Face's logo

Spaces:
amruthakp7
/
rock-classifier
Sleeping

App Files Community
1
ranjith commited on
Commit
1f40754
·
1 Parent(s): 70194b3

updated sd

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import re
5
 
6
 
7
 
8
- learn = load_learner('export.pkl')
9
 
10
  labels = learn.dls.vocab
11
  def predict(img):
 
5
 
6
 
7
 
8
+ learn = load_learner('rocks-2.pkl')
9
 
10
  labels = learn.dls.vocab
11
  def predict(img):