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 |
|