Spaces:
Runtime error
Runtime error
Commit
·
1185f42
1
Parent(s):
6076af7
categories ordered correctly
Browse files
app.py
CHANGED
|
@@ -2,9 +2,12 @@ import gradio as gr
|
|
| 2 |
from fastai.vision.all import *
|
| 3 |
|
| 4 |
# load the trained model
|
| 5 |
-
learn = load_learner('
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
|
| 7 |
-
categories = ['ironman','hulk','captain america','captain marvel','thanos','thor','vision','ant man','black panther','black widow','doctor strange','spiderman']
|
| 8 |
|
| 9 |
def classify_image(img):
|
| 10 |
pred, idx, probab = learn.predict(img)
|
|
|
|
| 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',
|
| 9 |
+
'doctor strange', 'hulk', 'ironman', 'spiderman', 'thanos', 'thor', 'vision']
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
def classify_image(img):
|
| 13 |
pred, idx, probab = learn.predict(img)
|