Spaces:

kaitehtzeng
/

primary_app

Sleeping

kaitehtzeng commited on Jan 28

Commit

c3ef27a

•

1 Parent(s): a70d1ae

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -133,7 +133,7 @@ class MeanPooling(nn.Module):
         return mean_embedding
 model = mymodel(config)
-model.load_state_dict(torch.load('my_model.pth'))
 model.eval()
 #preds = []

         return mean_embedding
 model = mymodel(config)
+model.load_state_dict(torch.load('my_model.pth',map_location=torch.device('cpu') ))
 model.eval()
 #preds = []