Update backend/main.py

backend/main.py

@@ -1,4 +1,5 @@
 from operator import index
+import os
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt
@@ -74,7 +75,9 @@ def predict(data : TransactionModel):
     return {"predictions": predictions.tolist()}
 
 
-if __name__ == "__main__":
-    uvicorn.run("main:app", host="0.0.0.0", port=8000)
+
+host = os.getenv("HOST", "0.0.0.0")
+port = os.getenv("PORT", 8000)
+uvicorn.run("main:app", host=host, port=port)