qweiq commited on
Commit
4683e7b
·
1 Parent(s): 14d50b5
Files changed (1) hide show
  1. app.py +1 -1
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('D:/Code/Antidote/ari_model/best.pth', map_location=torch.device('cpu'))
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