arikat commited on
Commit
feb7c28
1 Parent(s): a1a7f2c

remove path mistake

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ import zipfile
20
  import os
21
 
22
  # Load the model from the file
23
- with open('/home/aarya/Documents/paper3/family_labels.pkl', 'rb') as filefam:
24
  yfam = pickle.load(filefam)
25
 
26
  tokenizerfam = AutoTokenizer.from_pretrained("facebook/esm2_t12_35M_UR50D") #facebook/esm2_t33_650M_UR50D
@@ -31,7 +31,7 @@ device
31
  modelfam = EsmForSequenceClassification.from_pretrained("facebook/esm2_t12_35M_UR50D", num_labels=len(yfam.classes_))
32
  modelfam = modelfam.to('cpu')
33
 
34
- modelfam.load_state_dict(torch.load("/home/aarya/Documents/paper3/family.pth"))
35
  modelfam.eval()
36
 
37
  x_testfam = ["""MAEVLRTLAGKPKCHALRPMILFLIMLVLVLFGYGVLSPRSLMPGSLERGFCMAVREPDH
 
20
  import os
21
 
22
  # Load the model from the file
23
+ with open('family_labels.pkl', 'rb') as filefam:
24
  yfam = pickle.load(filefam)
25
 
26
  tokenizerfam = AutoTokenizer.from_pretrained("facebook/esm2_t12_35M_UR50D") #facebook/esm2_t33_650M_UR50D
 
31
  modelfam = EsmForSequenceClassification.from_pretrained("facebook/esm2_t12_35M_UR50D", num_labels=len(yfam.classes_))
32
  modelfam = modelfam.to('cpu')
33
 
34
+ modelfam.load_state_dict(torch.load("family.pth"))
35
  modelfam.eval()
36
 
37
  x_testfam = ["""MAEVLRTLAGKPKCHALRPMILFLIMLVLVLFGYGVLSPRSLMPGSLERGFCMAVREPDH