Spaces:
Runtime error
Runtime error
Topallaj Denis
commited on
Commit
·
84b598c
1
Parent(s):
d0ba84e
modified path to file
Browse files
main.py
CHANGED
@@ -57,8 +57,8 @@ class EndpointHandler():
|
|
57 |
self.model = model
|
58 |
|
59 |
# path to the vocab_content and trfm model
|
60 |
-
vocab_content_path =
|
61 |
-
trfm_path =
|
62 |
|
63 |
# load the vocab_content instead of the pickle file
|
64 |
with open(vocab_content_path, "r", encoding="utf-8") as f:
|
@@ -72,9 +72,9 @@ class EndpointHandler():
|
|
72 |
self.trfm.eval()
|
73 |
|
74 |
# path to the pretrained models
|
75 |
-
self.Km_model_path =
|
76 |
-
self.Kcat_model_path =
|
77 |
-
self.Kcat_over_Km_model_path =
|
78 |
|
79 |
# vocab indices
|
80 |
self.pad_index = 0
|
|
|
57 |
self.model = model
|
58 |
|
59 |
# path to the vocab_content and trfm model
|
60 |
+
vocab_content_path = "vocab_content.txt"
|
61 |
+
trfm_path = "trfm_12_23000.pkl"
|
62 |
|
63 |
# load the vocab_content instead of the pickle file
|
64 |
with open(vocab_content_path, "r", encoding="utf-8") as f:
|
|
|
72 |
self.trfm.eval()
|
73 |
|
74 |
# path to the pretrained models
|
75 |
+
self.Km_model_path = "Km.pkl"
|
76 |
+
self.Kcat_model_path = "Kcat.pkl"
|
77 |
+
self.Kcat_over_Km_model_path = "Kcat_over_Km.pkl"
|
78 |
|
79 |
# vocab indices
|
80 |
self.pad_index = 0
|