torinriley commited on
Commit
340599d
·
verified ·
1 Parent(s): 0229607

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -21,7 +21,7 @@ class EndpointHandler:
21
  self.device = device
22
 
23
  # Paths for weights and tokenizers
24
- self.model_weights_path = os.path.join(path, "SAVE.pt") # Ensure SAVE.pt is uploaded in the repository
25
  self.tokenizer_src_path = os.path.join(path, "tokenizer_en.json")
26
  self.tokenizer_tgt_path = os.path.join(path, "tokenizer_it.json")
27
 
 
21
  self.device = device
22
 
23
  # Paths for weights and tokenizers
24
+ self.model_weights_path = os.path.join(path, "EN-IT.pt")
25
  self.tokenizer_src_path = os.path.join(path, "tokenizer_en.json")
26
  self.tokenizer_tgt_path = os.path.join(path, "tokenizer_it.json")
27