Update app.py

app.py

@@ -15,15 +15,14 @@ try:
 
     class GenerateRequest(BaseModel):
         prompt: str
+        key: str
 
     class GenerateResponse(BaseModel):
         generated_text: str
 
     @app.post("/generate", response_model=GenerateResponse)
-    async def generate(request: GenerateRequest, authorization: str = Header(None)):
-        if not authorization or not authorization.startswith('Bearer '):
-            raise HTTPException(status_code=401, detail="Unauthorized")
-
+    async def generate(request: GenerateRequest):
+        authorization = request.key
         token = authorization.split('Bearer ')[1]
         if token != API_KEY:
             raise HTTPException(status_code=401, detail="Unauthorized")