Spaces:
Sleeping
Sleeping
App.py model force cpu use
Browse files
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 = []
|