Spaces:

kaitehtzeng
/

primary_app

Sleeping

kaitehtzeng commited on Jan 28

Commit

81cd130

•

1 Parent(s): 6bc9b7f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -132,7 +132,7 @@ class MeanPooling(nn.Module):
         return mean_embedding
-model = mymodel(config).to(map_location=torch.device('cpu'))
 model.load_state_dict(torch.load('my_model.pth'))
 model.eval()

         return mean_embedding
+model = mymodel(config)
 model.load_state_dict(torch.load('my_model.pth'))
 model.eval()