Update handler.py

handler.py

@@ -1,9 +1,15 @@
 import torch
 from typing import Dict, List, Any
 from tokenizers import Tokenizer
-from model import build_transformer
+import sys
+import os
 import warnings
 
+# Add the current directory to the system path to locate the model module
+sys.path.append(os.path.dirname(__file__))
+
+from model import build_transformer
+
 warnings.simplefilter("ignore", category=FutureWarning)
 
 class EndpointHandler:
@@ -14,10 +20,10 @@ class EndpointHandler:
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.device = device
 
-        # Path to weights and tokenizers
-        self.model_weights_path = path + "/SAVE.pt"  # Ensure SAVE.pt is uploaded in the repository
-        self.tokenizer_src_path = path + "/tokenizer_en.json"
-        self.tokenizer_tgt_path = path + "/tokenizer_it.json"
+        # Paths for weights and tokenizers
+        self.model_weights_path = os.path.join(path, "SAVE.pt")  # Ensure SAVE.pt is uploaded in the repository
+        self.tokenizer_src_path = os.path.join(path, "tokenizer_en.json")
+        self.tokenizer_tgt_path = os.path.join(path, "tokenizer_it.json")
 
         # Load tokenizers
         self.tokenizer_src = Tokenizer.from_file(self.tokenizer_src_path)