Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Depends
+from fastapi import FastAPI, Depends, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
@@ -61,10 +61,7 @@ async def generate_text(item: Item, token: Annotated[str, Depends(oauth2_scheme)
     # Reject if not authenticated
     apiKey = os.environ.get("API_KEY")
     if apiKey != token:
-        return {
-            "message": "Not authenticated",
-            "token": token
-        }
+        raise HTTPException(status_code=403, detail="Invalid API key")
 
     return {
         "response": generate(item)