Update main.py

main.py

@@ -34,10 +34,10 @@ async def api_set_config(args: dict):
 @app.post("/infer")
 async def api_infer(args: dict):
     args['fastapi'] = True
-    content = fn.infer(args['input'], [], args['instruct'], args)
+    content = fn.infer(args['input'], [], args['instruct'], None, args)
     return {'content': content}
 
 @app.post("/numel")
 async def api_numel(args: dict):
-    content = fn.numel(args['input'], [], args['instruct'], args)
+    content = fn.numel(args['input'], [], args['instruct'], None, args)
     return {'numel': content}