Update app.py

app.py

@@ -1,19 +1,14 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
+from contextlib import asynccontextmanager
 import os
 
-# Only import transformers when we need it
-app = FastAPI(title="Quiz Guru Chatbot", version="1.0.0")
-
 # Global variables
 model = None
 tokenizer = None
 device = None
 model_loaded = False
 
-class PromptRequest(BaseModel):
-    prompt: str
-
 def load_model():
     global model, tokenizer, device, model_loaded
     try:
@@ -46,11 +41,19 @@ def load_model():
         print(f"❌ Error loading model: {e}")
         return False
 
-@app.on_event("startup")
-async def startup_event():
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup
     print("🚀 Starting up...")
-    # Don't block startup if model fails to load
     load_model()
+    yield
+    # Shutdown (if needed)
+    print("🛑 Shutting down...")
+
+app = FastAPI(title="Quiz Guru Chatbot", version="1.0.0", lifespan=lifespan)
+
+class PromptRequest(BaseModel):
+    prompt: str
 
 @app.get("/")
 def read_root():