Update app.py

app.py

@@ -48,7 +48,7 @@ class ModelPre(torch.nn.Module):
 # Load the pretrained model 
 model = ModelPre()
 
-model_w = torch.load("SVD/GeoG.pth", map_location=torch.device(device))
+model_w = xser.load("SVD/GeoG.pth")['model'].to(device)
 model.load_state_dict(model_w['model'])
 
 cmp = transforms.Compose([