Update api/main.py

api/main.py

@@ -13,6 +13,10 @@ import re
 import shap
 import numpy as np
 
+from pathlib import Path
+
+
+
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 os.chdir(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
@@ -28,9 +32,16 @@ app = FastAPI(
     version="2.0.0"
 )
 
+app = FastAPI(title="PlateVision Backend")
+
+
+FRONTEND_URL = os.environ.get("FRONTEND_URL")
+
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["http://localhost:5173"],
+    allow_origins=[
+        FRONTEND_URL
+                  ],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],