slacktoid commited on
Commit
53dc593
·
verified ·
1 Parent(s): 2ec5173

Update handler.py

Browse files

changed class name from Handler to EndpointHandler

Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -1,7 +1,7 @@
1
  # handler.py
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
- class Handler:
5
  def __init__(self, path=""):
6
  self.tokenizer = AutoTokenizer.from_pretrained(path)
7
  self.model = AutoModelForCausalLM.from_pretrained(path)
 
1
  # handler.py
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
+ class EndpointHandler:
5
  def __init__(self, path=""):
6
  self.tokenizer = AutoTokenizer.from_pretrained(path)
7
  self.model = AutoModelForCausalLM.from_pretrained(path)