Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException, Depends, status
-from fastapi.security import HTTPBearer, HTTPBearerToken
+from fastapi.security import HTTPBearer
 from pydantic import BaseModel
 from transformers import pipeline
 import gradio as gr
@@ -34,13 +34,13 @@ class EnhanceRequest(BaseModel):
 class EnhanceResponse(BaseModel):
     enhanced_prompt: str
 
-def verify_api_key(token: HTTPBearerToken = Depends(security)):
-    if token.credentials != API_KEY:
+def verify_api_key(credentials = Depends(security)):
+    if credentials.credentials != API_KEY:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
             detail="Invalid API key"
         )
-    return token.credentials
+    return credentials.credentials
 
 @app.post("/enhance", response_model=EnhanceResponse)
 async def enhance_prompt(request: EnhanceRequest, api_key: str = Depends(verify_api_key)):