RASMUS commited on
Commit
53abc6e
·
verified ·
1 Parent(s): 635557f

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -10,7 +10,7 @@ def generate_rag_prompt_message(context, question):
10
  return prompt
11
 
12
 
13
- class PreTrainedPipeline():
14
  def __init__(self, path=""):
15
  # Preload all the elements you are going to need at inference.
16
  # pseudo:
 
10
  return prompt
11
 
12
 
13
+ class EndpointHandler():
14
  def __init__(self, path=""):
15
  # Preload all the elements you are going to need at inference.
16
  # pseudo: