Hugging Face's logo

Spaces:
Gaeulove924
/
testforclass
Runtime error

App Files Community
Gaeulove924 commited on
Commit
9e4538e
•
1 Parent(s): 718d11e

Update app.py

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