point to local file

handler.py

@@ -1,7 +1,7 @@
 from typing import Dict, List, Any
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 import torch
-from handler import EndpointHandler
+
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -11,7 +11,7 @@ class EndpointHandler():
         
 
         # Load the fine-tuned model
-        self.date_model_path = "deberta-qa-finetuned"
+        self.date_model_path = "./deberta-qa-finetuned"
         self.date_tokenizer = AutoTokenizer.from_pretrained(self.date_model_path)
         self.date_model = AutoModelForQuestionAnswering.from_pretrained(self.date_model_path)
 
@@ -75,7 +75,7 @@ class EndpointHandler():
         return answer
 
 
-
+from handler import EndpointHandler
 
 # init handler
 my_handler = EndpointHandler(path=".")
@@ -89,5 +89,5 @@ non_holiday_pred=my_handler(non_holiday_payload)
 # holiday_payload=my_handler(holiday_payload)
 
 # show results
-# print("non_holiday_pred", non_holiday_pred)
+print(non_holiday_pred)
 # print("holiday_payload", holiday_payload)