Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -85,7 +85,7 @@ class InferencePipeline():
|
|
85 |
class EndpointHandler():
|
86 |
def __init__(self):
|
87 |
self.inference_pipeline = InferencePipeline()
|
88 |
-
self.pipeline = pipeline("
|
89 |
|
90 |
def __call__(self, question):
|
91 |
answer = self.inference_pipeline.get_answer_e2e(question)
|
|
|
85 |
class EndpointHandler():
|
86 |
def __init__(self):
|
87 |
self.inference_pipeline = InferencePipeline()
|
88 |
+
self.pipeline = pipeline("question-answering", model=self.inference_pipeline, tokenizer=None)
|
89 |
|
90 |
def __call__(self, question):
|
91 |
answer = self.inference_pipeline.get_answer_e2e(question)
|