Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -53,6 +53,7 @@ def verify_token(x_api_key: str = Depends(token_header)):
|
|
53 |
raise HTTPException(status_code=403, detail="Unauthorized")
|
54 |
return x_api_key
|
55 |
|
|
|
56 |
@app.post("/ner")
|
57 |
@limiter.limit("10/minute")
|
58 |
def ner_predict(request: TextRequest, api_key: str = Depends(verify_token), req: Request = None):
|
@@ -64,6 +65,27 @@ def ner_predict(request: TextRequest, api_key: str = Depends(verify_token), req:
|
|
64 |
"entities": predictions,
|
65 |
"usage": API_KEYS[api_key]["usage_count"]
|
66 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
67 |
|
68 |
@app.post("/register_user")
|
69 |
def register_user(request: RegisterRequest, x_api_key: str = Depends(token_header)):
|
|
|
53 |
raise HTTPException(status_code=403, detail="Unauthorized")
|
54 |
return x_api_key
|
55 |
|
56 |
+
"""
|
57 |
@app.post("/ner")
|
58 |
@limiter.limit("10/minute")
|
59 |
def ner_predict(request: TextRequest, api_key: str = Depends(verify_token), req: Request = None):
|
|
|
65 |
"entities": predictions,
|
66 |
"usage": API_KEYS[api_key]["usage_count"]
|
67 |
}
|
68 |
+
"""
|
69 |
+
|
70 |
+
from fastapi import Request
|
71 |
+
|
72 |
+
@app.post("/ner")
|
73 |
+
@limiter.limit("10/minute")
|
74 |
+
def ner_predict(
|
75 |
+
body: TextRequest,
|
76 |
+
req: Request,
|
77 |
+
api_key: str = Depends(verify_token)
|
78 |
+
):
|
79 |
+
logger.info("NER request from IP: %s", get_remote_address(req))
|
80 |
+
predictions = ner_pipeline(body.text)
|
81 |
+
API_KEYS[api_key]["usage_count"] = API_KEYS[api_key].get("usage_count", 0) + 1
|
82 |
+
save_keys()
|
83 |
+
return {
|
84 |
+
"entities": predictions,
|
85 |
+
"usage": API_KEYS[api_key]["usage_count"]
|
86 |
+
}
|
87 |
+
|
88 |
+
|
89 |
|
90 |
@app.post("/register_user")
|
91 |
def register_user(request: RegisterRequest, x_api_key: str = Depends(token_header)):
|