Hugging Face's logo

Spaces:
Stevenqaq
/
image-classifdier
Sleeping

App Files Community
Stevenqaq commited on
Commit
da6d75b
·
verified ·
1 Parent(s): 652aff2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def predict(img):
20
 
21
 
22
  path = "./"
23
- learn_inf = load_learner(path + "demo_model.pkl")
24
 
25
  option = st.radio("", ["Upload Image", "Image URL"])
26
 
 
20
 
21
 
22
  path = "./"
23
+ learn_inf = load_learner(path + "resnet34_stage-1.pkl")
24
 
25
  option = st.radio("", ["Upload Image", "Image URL"])
26