Spaces:
Sleeping
Sleeping
minor edits
Browse files
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("
|
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('
|
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)
|