Update main.py

main.py

@@ -11,6 +11,8 @@ from fastapi.responses import FileResponse, HTMLResponse
 from pydantic import BaseModel
 
 import functools
+import math
+import numpy as np
 
 
 
@@ -26,8 +28,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-import math
-import numpy as np
+FASTF1_CACHE_DIR = os.environ['FASTF1_CACHE_DIR']
+
+
 def smooth_derivative(t_in, v_in):
 
     #
@@ -249,7 +252,7 @@ def sessions_available(year: int, event: str | int) -> any:
 @functools.cache
 @app.get("/{year}/{event}/{session}", response_model=None)
 def session_drivers(year: int, event: str | int, session: str) -> any:
-    # fastf1.Cache.enable_cache('cache')
+    fastf1.Cache.enable_cache(FASTF1_CACHE_DIR)
     # get drivers available for a given year, event and session
     f1session = fastf1.get_session(year, event, session)
     api_path = f1session.api_path