Hugging Face's logo

Spaces:
robocan
/
GeoGuessrRobot
Running

App Files Community
robocan commited on
Commit
d497c1d
·
verified ·
1 Parent(s): b7d3da6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ class ModelPre(torch.nn.Module):
49
  # Load the pretrained model
50
  model = ModelPre()
51
 
52
- model_w = xser.load("SVD/GeoG.pth")['model'].to(device)
53
  model.load_state_dict(model_w['model'])
54
 
55
  cmp = transforms.Compose([
 
49
  # Load the pretrained model
50
  model = ModelPre()
51
 
52
+ model_w = xser.load("SVD/GeoG.pth")
53
  model.load_state_dict(model_w['model'])
54
 
55
  cmp = transforms.Compose([