Spaces:

meghnabjayakar
/

drug-target-interaction-model

Sleeping

meghnabjayakar commited on Oct 19, 2024

Commit

1b00e9c

verified ·

1 Parent(s): 9414524

final update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ tokenizer = BertTokenizer.from_pretrained("dmis-lab/biobert-base-cased-v1.1")
 model = BertForSequenceClassification.from_pretrained("dmis-lab/biobert-base-cased-v1.1", num_labels=2)
 # loading the pretrained weights into the model
-model.load_state_dict(torch.load('Bio_BERT_model.pth'))
 device = "cpu"

 model = BertForSequenceClassification.from_pretrained("dmis-lab/biobert-base-cased-v1.1", num_labels=2)
 # loading the pretrained weights into the model
+model.load_state_dict(torch.load('Bio_BERT_model.pth', map_location=torch.device('cpu')))
 device = "cpu"