Topallaj Denis commited on
Commit
84b598c
·
1 Parent(s): d0ba84e

modified path to file

Browse files
Files changed (1) hide show
  1. main.py +5 -5
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 = f"{path}/vocab_content.txt"
61
- trfm_path = f"{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,9 +72,9 @@ class EndpointHandler():
72
  self.trfm.eval()
73
 
74
  # path to the pretrained models
75
- self.Km_model_path = f"{path}/Km.pkl"
76
- self.Kcat_model_path = f"{path}/Kcat.pkl"
77
- self.Kcat_over_Km_model_path = f"{path}/Kcat_over_Km.pkl"
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