shin-mashita
commited on
Commit
•
eb8dc27
1
Parent(s):
0176f1f
Minor edits
Browse files
app.py
CHANGED
|
@@ -44,6 +44,7 @@ def classify(video,dataset='WLASL100'):
|
|
| 44 |
input = preprocess(video)
|
| 45 |
|
| 46 |
model = InceptionI3d()
|
|
|
|
| 47 |
model.load_state_dict(torch.load('weights/rgb_imagenet.pt'))
|
| 48 |
model.replace_logits(to_load[dataset]['logits'])
|
| 49 |
model.load_state_dict(torch.load(to_load[dataset]['path']))
|
|
|
|
| 44 |
input = preprocess(video)
|
| 45 |
|
| 46 |
model = InceptionI3d()
|
| 47 |
+
model.cpu()
|
| 48 |
model.load_state_dict(torch.load('weights/rgb_imagenet.pt'))
|
| 49 |
model.replace_logits(to_load[dataset]['logits'])
|
| 50 |
model.load_state_dict(torch.load(to_load[dataset]['path']))
|