Hugging Face's logo

Spaces:
manish-pro
/
dL_avengers
Runtime error

App Files Community
manish-pro commited on
Commit
8d9c7de
·
1 Parent(s): 1185f42

model name update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from fastai.vision.all import *
3
 
4
  # load the trained model
5
- learn = load_learner('model_one.pkl')
6
 
7
  # learn.dls.vocab where dls in the dataloaders from DataBlock
8
  categories = ['ant man', 'black panther', 'black widow', 'captain america', 'captain marvel',
 
2
  from fastai.vision.all import *
3
 
4
  # load the trained model
5
+ learn = load_learner('model.pkl')
6
 
7
  # learn.dls.vocab where dls in the dataloaders from DataBlock
8
  categories = ['ant man', 'black panther', 'black widow', 'captain america', 'captain marvel',