arikat commited on
Commit
d341fbe
1 Parent(s): 2b90f79

minor edits

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ device
33
  modelfam = EsmForSequenceClassification.from_pretrained("facebook/esm2_t33_650M_UR50D", num_labels=len(set(labelsfam.tolist())))
34
  modelfam = modelfam.to('cpu')
35
 
36
- modelfam.load_state_dict(torch.load("/home/aarya/Documents/paper3/model_650M.pth"))
37
  modelfam.eval()
38
 
39
  x_testfam = ["""MAEVLRTLAGKPKCHALRPMILFLIMLVLVLFGYGVLSPRSLMPGSLERGFCMAVREPDH
@@ -63,7 +63,7 @@ decoded_labelsfam
63
  #Load donor model from file
64
  tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t12_35M_UR50D")
65
 
66
- with open('donorslabels.pkl', 'rb') as file:
67
  label_encoder = pickle.load(file)
68
 
69
  # encoded_labels = label_encoder.fit(y)
 
33
  modelfam = EsmForSequenceClassification.from_pretrained("facebook/esm2_t33_650M_UR50D", num_labels=len(set(labelsfam.tolist())))
34
  modelfam = modelfam.to('cpu')
35
 
36
+ modelfam.load_state_dict(torch.load("model_650M.pth"))
37
  modelfam.eval()
38
 
39
  x_testfam = ["""MAEVLRTLAGKPKCHALRPMILFLIMLVLVLFGYGVLSPRSLMPGSLERGFCMAVREPDH
 
63
  #Load donor model from file
64
  tokenizer = AutoTokenizer.from_pretrained("facebook/esm2_t12_35M_UR50D")
65
 
66
+ with open('donor_labels.pkl', 'rb') as file:
67
  label_encoder = pickle.load(file)
68
 
69
  # encoded_labels = label_encoder.fit(y)