Hugging Face's logo

Spaces:
guumaster
/
fastai-classifier

App Files Community
guumaster commited on
Commit
5decb05
•
1 Parent(s): f88e7a8

change to resnet50

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
 
4
 
5
  # Cell
6
- learn = load_learner("model-resnet18.pkl")
7
 
8
  # Cell
9
  categories = learn.dls.vocab
 
3
 
4
 
5
  # Cell
6
+ learn = load_learner("model-resnet50.pkl")
7
 
8
  # Cell
9
  categories = learn.dls.vocab