johnpaulbin commited on
Commit
1877179
1 Parent(s): c769c97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -100,7 +100,7 @@ model = nn.Sequential(
100
  nn.Dropout(0.05) # Dropout
101
  )
102
 
103
- model.load_state_dict(torch.load("large.pth"))
104
  model.eval()
105
 
106
  @app.route('/infer', methods=['POST'])
 
100
  nn.Dropout(0.05) # Dropout
101
  )
102
 
103
+ model.load_state_dict(torch.load("large.pth"), map_location=torch.device('cpu'))
104
  model.eval()
105
 
106
  @app.route('/infer', methods=['POST'])