Spaces:

meghnabjayakar
/

drug-target-interaction-model

Running

meghnabjayakar commited on Oct 19, 2024

Commit

7407ba5

verified ·

1 Parent(s): 4f5a32f

final updates

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,9 @@ 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', map_location=torch.device('cpu')))
-print("Done loading the weights!")
 device = "cpu"

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