Update app.py

app.py

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