Update app.py

app.py

@@ -1,14 +1,16 @@
 from fastapi import FastAPI, HTTPException
 import base64
 import os
+from contextlib import asynccontextmanager
 
 app = FastAPI()
 
 # Global variable to hold the LoRA weights
 lora_weights = None
 
-@app.on_event("startup")
-async def load_lora_weights():
+# Lifespan event to load the LoRA weights
+@asynccontextmanager
+async def lifespan(app: FastAPI):
     global lora_weights
     lora_path = "./lora_file.pth"  # Ensure the correct file name
     if os.path.exists(lora_path):
@@ -17,6 +19,10 @@ async def load_lora_weights():
         print("LoRA weights loaded and preprocessed successfully.")
     else:
         raise HTTPException(status_code=500, detail="LoRA file not found.")
+    yield
+    # Cleanup if necessary (but not required in this case)
+
+app.add_event_handler("lifespan", lifespan)
 
 @app.post("/modify-prompt")
 async def modify_prompt(prompt: str):