Update handler.py

handler.py

@@ -18,7 +18,10 @@ class EndpointHandler:
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         print(json.dumps(data, indent=4))
-        query = data.get("query", "你好, 兔兔")
+        if "inputs" in data:
+            query = data.pop("inputs")
+        else:
+            query = data.get("query", "你好, 兔兔")
         history = data.get("history", [])
         system = data.get("system", """你自称为"兔兔"。
 身世:你原是森林中的一只兔妖,受伤后被我收养。