Upload app.py

app.py

@@ -71,12 +71,8 @@ from fastapi import Request
 
 @app.post("/ner")
 @limiter.limit("10/minute")
-def ner_predict(
-    body: TextRequest, 
-    req: Request, 
-    api_key: str = Depends(verify_token)
-):
-    logger.info("NER request from IP: %s", get_remote_address(req))
+def ner_predict(body: TextRequest, request: Request, api_key: str = Depends(verify_token)):
+    logger.info("NER request from IP: %s", get_remote_address(request))
     predictions = ner_pipeline(body.text)
     API_KEYS[api_key]["usage_count"] = API_KEYS[api_key].get("usage_count", 0) + 1
     save_keys()