Update app.py

app.py

@@ -3,12 +3,10 @@ import base64
 import os
 from contextlib import asynccontextmanager
 
-app = FastAPI()
-
 # Global variable to hold the LoRA weights
 lora_weights = None
 
-# Lifespan event to load the LoRA weights
+# Lifespan context manager to handle the application startup
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     global lora_weights
@@ -22,7 +20,8 @@ async def lifespan(app: FastAPI):
     yield
     # Cleanup if necessary (but not required in this case)
 
-app.add_event_handler("lifespan", lifespan)
+# Initialize FastAPI app with lifespan context
+app = FastAPI(lifespan=lifespan)
 
 @app.post("/modify-prompt")
 async def modify_prompt(prompt: str):