Spaces:
Sleeping
Sleeping
remove path mistake
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ import zipfile
|
|
20 |
import os
|
21 |
|
22 |
# Load the model from the file
|
23 |
-
with open('
|
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("
|
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
|