Update handler.py

handler.py

@@ -4,9 +4,13 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
 from transformers import TrainingArguments, Trainer
 import torch
 
+import os
+model_dir2 = os.path.abspath("./json_extraction_all")
+
+
 class EndpointHandler:
     def __init__(self, model_dir):
-        self.model_dir = model_dir
+        self.model_dir = model_dir2
         self.model = None
         self.tokenizer = None
         self.load()  # Ensure model loads on initialization