meghnabjayakar commited on
Commit
4f5a32f
·
verified ·
1 Parent(s): 1b00e9c

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,6 +10,7 @@ model = BertForSequenceClassification.from_pretrained("dmis-lab/biobert-base-cas
10
 
11
  # loading the pretrained weights into the model
12
  model.load_state_dict(torch.load('Bio_BERT_model.pth', map_location=torch.device('cpu')))
 
13
 
14
  device = "cpu"
15
 
@@ -107,7 +108,7 @@ def display_image():
107
 
108
 
109
  title = "Unlocking Precision: Predicting Drug-Target Interactions for Safer, Smarter Treatments"
110
- article = "Stay Safe!"
111
  examples = [
112
  ["Targeting the secretion of sEV PD-L1 has emerged as a promising strategy to enhance immunotherapy"],
113
  ["These results suggested that AM would be a useful platform for the development of a new radiopharmaceutical targeting ER"],
 
10
 
11
  # loading the pretrained weights into the model
12
  model.load_state_dict(torch.load('Bio_BERT_model.pth', map_location=torch.device('cpu')))
13
+ print("Done loading the weights!")
14
 
15
  device = "cpu"
16
 
 
108
 
109
 
110
  title = "Unlocking Precision: Predicting Drug-Target Interactions for Safer, Smarter Treatments"
111
+ article = "Stay Safe! Project by MBJ @24"
112
  examples = [
113
  ["Targeting the secretion of sEV PD-L1 has emerged as a promising strategy to enhance immunotherapy"],
114
  ["These results suggested that AM would be a useful platform for the development of a new radiopharmaceutical targeting ER"],