Linsad commited on
Commit
631559b
1 Parent(s): 816f0f4

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +5 -3
handler.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from typing import Dict, List, Any
2
  from transformers import AutoTokenizer, AutoModel
3
 
@@ -17,8 +18,9 @@ class EndpointHandler:
17
  A :obj:`list` | `dict`: will be serialized and returned
18
  """
19
  # get inputs
20
- inputs = data.pop("inputs", data)
 
21
 
22
- response, history = self.model.chat(self.tokenizer, inputs, history=[])
23
 
24
- return [{'response': response, 'history': history}]
 
1
+ import os
2
  from typing import Dict, List, Any
3
  from transformers import AutoTokenizer, AutoModel
4
 
 
18
  A :obj:`list` | `dict`: will be serialized and returned
19
  """
20
  # get inputs
21
+ return str(os.getcwd())
22
+ # inputs = data.pop("inputs", data)
23
 
24
+ # response, history = self.model.chat(self.tokenizer, inputs, history=[])
25
 
26
+ # return [{'response': response, 'history': history}]