Update app.py

app.py

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