foxxy-hm commited on
Commit
36adcd4
·
1 Parent(s): 01b8cc7

Update handler.py

Browse files
Files changed (1) hide show
  1. 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("qa-model", model=self.inference_pipeline, tokenizer=None)
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)