Update server/main.py

justinthelaw · Aug 7, 2023 · e7edc84 · e7edc84
1 parent 2dd7451
commit e7edc84
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/server/main.py b/server/main.py
@@ -1,12 +1,20 @@
 from fastapi import FastAPI
 from transformers import logging
+from dotenv import load_dotenv
+from uvicorn import run
+import os
 
 from src.controllers.ForgeController import ForgeController
 
 app = FastAPI()
 
 logging.set_verbosity_error()
 
+load_dotenv("../config/.env.local")
+
+host = os.getenv("SERVER_HOST")
+port = int(os.getenv("SERVER_PORT"))
+
 
 @app.get("/")
 def welcome():
@@ -22,3 +30,5 @@ def welcome():
 forge_routes = ForgeController().get_router()
 
 app.include_router(forge_routes)
+
+run("main:app", host=host, port=port)