ejqs
/

BK_QNADateExtractor

Question Answering

Model card Files Files and versions

earlsab commited on Mar 29, 2025

Commit

2d6f3d7

·

1 Parent(s): cd1f8e3

Add path

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -11,7 +11,7 @@ class EndpointHandler():
         # Load the fine-tuned model
-        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)

         # Load the fine-tuned model
+        self.date_model_path = path + "/deberta-qa-finetuned"
         self.date_tokenizer = AutoTokenizer.from_pretrained(self.date_model_path)
         self.date_model = AutoModelForQuestionAnswering.from_pretrained(self.date_model_path)