Padzong commited on
Commit
52b0d10
·
verified ·
1 Parent(s): 41b4975

App.py model force cpu use

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,10 +7,10 @@ import torch, os
7
  import torch.nn.functional as F
8
 
9
  model = Siamese_nn()
10
- weights = torch.load('trained_model')
11
  model.load_state_dict(weights)
12
  model.eval()
13
- #%%
14
  file_list = os.listdir('data')
15
  examples = []
16
  usersIndexes = []
 
7
  import torch.nn.functional as F
8
 
9
  model = Siamese_nn()
10
+ weights = torch.load('trained_model', map_location=torch.device('cpu'))
11
  model.load_state_dict(weights)
12
  model.eval()
13
+
14
  file_list = os.listdir('data')
15
  examples = []
16
  usersIndexes = []