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

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -85,8 +85,8 @@ class InferencePipeline():
85
  class EndpointHandler():
86
  def __init__(self):
87
  self.inference_pipeline = InferencePipeline()
88
- pipeline = pipeline("qa-model", model=self.inference_pipeline, tokenizer=None)
89
 
90
  def __call__(self, question):
91
- answer = self.endpoint_handler.get_answer_e2e(question)
92
  return answer
 
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)
92
  return answer