Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
@@ -238,7 +238,12 @@ tokenizer = AutoTokenizer.from_pretrained(config['tokenizer_path'])
|
|
238 |
path = 'model.pth'
|
239 |
|
240 |
model = NBMEModel().to(config['device'])
|
241 |
-
model.load_state_dict(
|
|
|
|
|
|
|
|
|
|
|
242 |
model.eval()
|
243 |
|
244 |
# input_text = create_sample_test()
|
|
|
238 |
path = 'model.pth'
|
239 |
|
240 |
model = NBMEModel().to(config['device'])
|
241 |
+
model.load_state_dict(
|
242 |
+
torch.load(
|
243 |
+
path,
|
244 |
+
map_location=torch.device(config['device'])
|
245 |
+
)
|
246 |
+
)
|
247 |
model.eval()
|
248 |
|
249 |
# input_text = create_sample_test()
|