Lin0He commited on
Commit
77147a1
·
1 Parent(s): d987ebd

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -3
handler.py CHANGED
@@ -74,10 +74,10 @@ def predict(text, model, tokenizer):
74
  #print("summary:", sorted(result_text, key=len)[3])
75
 
76
  class EndpointHandler():
77
- def __init__(self, path="Lin0He/text-summary-gpt2-short"):
78
  # load model and tokenizer from path
79
- self.tokenizer = AutoTokenizer.from_pretrained(path)
80
- self.model = AutoModelForCausalLM.from_pretrained(path)
81
 
82
 
83
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
 
74
  #print("summary:", sorted(result_text, key=len)[3])
75
 
76
  class EndpointHandler():
77
+ def __init__(self, path=""):
78
  # load model and tokenizer from path
79
+ self.tokenizer = AutoTokenizer.from_pretrained("Lin0He/text-summary-gpt2-short")
80
+ self.model = AutoModelForCausalLM.from_pretrained("Lin0He/text-summary-gpt2-short")
81
 
82
 
83
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]: