Itamarl commited on
Commit
b1e6d73
·
1 Parent(s): eab9373

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -1
handler.py CHANGED
@@ -55,5 +55,7 @@ class EndpointHandler():
55
 
56
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
57
  logger.info(f"iiinnnnnnnnnn {data}")
58
- res = self.generate_text(data)
 
 
59
  return res
 
55
 
56
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
57
  logger.info(f"iiinnnnnnnnnn {data}")
58
+ inputs = data.pop("inputs",data)
59
+ logger.info(f"iiinnnnnnnnnnbbbbbb {inputs}")
60
+ res = self.generate_text(inputs)
61
  return res