hassoudi commited on
Commit
75c8918
·
verified ·
1 Parent(s): e1ec338

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -0
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)):