simplify code

Former-commit-id: d3731754ab7c28ae81f60784e0e4213f279d93fe
2023-07-20 15:08:57 +08:00
parent 7a43ff3d89
commit 5ba0b80e5c
18 changed files with 52 additions and 136 deletions
--- a/src/api_demo.py
+++ b/src/api_demo.py
@@ -5,9 +5,13 @@

 import uvicorn

-from llmtuner import create_app
+from llmtuner.api.app import create_app
+
+
+def main():
+    app = create_app()
+    uvicorn.run(app, host="0.0.0.0", port=8000, workers=1)


 if __name__ == "__main__":
-    app = create_app()
-    uvicorn.run(app, host="0.0.0.0", port=8000, workers=1)
+    main()