Spaces:

cedssama
/

I3D_Sign_Language_Classification

Runtime error

cedssama commited on Mar 26, 2022

Commit

eb8dc27

1 Parent(s): 0176f1f

Minor edits

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,6 +44,7 @@ def classify(video,dataset='WLASL100'):
     input = preprocess(video)
     model = InceptionI3d()
     model.load_state_dict(torch.load('weights/rgb_imagenet.pt'))
     model.replace_logits(to_load[dataset]['logits'])
     model.load_state_dict(torch.load(to_load[dataset]['path']))

     input = preprocess(video)
     model = InceptionI3d()
+    model.cpu()
     model.load_state_dict(torch.load('weights/rgb_imagenet.pt'))
     model.replace_logits(to_load[dataset]['logits'])
     model.load_state_dict(torch.load(to_load[dataset]['path']))