new
Browse files
app.py
CHANGED
|
@@ -123,7 +123,7 @@ def model_infer(input_a, input_b):
|
|
| 123 |
'michiyasunaga/BioLinkBERT-base',
|
| 124 |
config=config,
|
| 125 |
)
|
| 126 |
-
p_sum = torch.load('
|
| 127 |
model.load_state_dict(p_sum)
|
| 128 |
tokenizer = AutoTokenizer.from_pretrained('michiyasunaga/BioLinkBERT-base')
|
| 129 |
|
|
|
|
| 123 |
'michiyasunaga/BioLinkBERT-base',
|
| 124 |
config=config,
|
| 125 |
)
|
| 126 |
+
p_sum = torch.load('best.pth', map_location=torch.device('cpu'))
|
| 127 |
model.load_state_dict(p_sum)
|
| 128 |
tokenizer = AutoTokenizer.from_pretrained('michiyasunaga/BioLinkBERT-base')
|
| 129 |
|