Update handler.py
Browse files- 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 |
-
|
|
|
|
|
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
|