torinriley
commited on
Update handler.py
Browse files- 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, "
|
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 |
|