johnpaulbin commited on
Commit
352195e
1 Parent(s): f2f7e61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ model = nn.Sequential(
98
  nn.Dropout(0.05) # Dropout
99
  )
100
 
101
- model.load_state_dict(torch.load("large.pth"))
102
  model.eval()
103
 
104
 
 
98
  nn.Dropout(0.05) # Dropout
99
  )
100
 
101
+ model.load_state_dict(torch.load("large.pth"), map_location=torch.device('cpu'))
102
  model.eval()
103
 
104