TymaaHammouda commited on
Commit
4e21739
·
verified ·
1 Parent(s): 2347550

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -0
app.py ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from huggingface_hub import hf_hub_download
3
+ import os
4
+ from pydantic import BaseModel
5
+ from fastapi.responses import JSONResponse
6
+
7
+ print("Version ---- 1")
8
+ app = FastAPI()
9
+
10
+ from sinatools.relations.relation_extractor import relation_extraction
11
+ from sinatools.relations.event_relation_extractor import event_argument_relation_extraction
12
+
13
+ class RelationRequest(BaseModel):
14
+ text: str
15
+
16
+ @app.post("/predict_relation")
17
+ def predict_relation(request: RelationRequest):
18
+ text = request.text
19
+
20
+ result = relation_extraction(sentence)
21
+
22
+ content = {"resp": result, "statusText": "OK", "statusCode": 0}
23
+
24
+ return JSONResponse(
25
+ content=content,
26
+ media_type="application/json",
27
+ status_code=200,
28
+ )
29
+
30
+
31
+
32
+ @app.post("/predict_event")
33
+ def predict_event(request: RelationRequest):
34
+ text = request.text
35
+
36
+ result = event_argument_relation_extraction(sentence)
37
+
38
+ content = {"resp": result, "statusText": "OK", "statusCode": 0}
39
+
40
+ return JSONResponse(
41
+ content=content,
42
+ media_type="application/json",
43
+ status_code=200,
44
+ )
45
+
46
+