Spaces:
Sleeping
Sleeping
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from fastapi import FastAPI, Request
|
2 |
+
from pydantic import BaseModel
|
3 |
+
from typing import List
|
4 |
+
from embedder import embed_clauses, embed_chunks
|
5 |
+
from tagger import clause_match_by_cosine
|
6 |
+
|
7 |
+
app = FastAPI()
|
8 |
+
|
9 |
+
class ChunkRequest(BaseModel):
|
10 |
+
chunks: List[str]
|
11 |
+
clause_reference: List[dict]
|
12 |
+
|
13 |
+
@app.post("/tag_clauses")
|
14 |
+
def tag_clauses(data: ChunkRequest):
|
15 |
+
embedded_chunks = embed_chunks(data.chunks)
|
16 |
+
embedded_references = embed_clauses(data.clause_reference)
|
17 |
+
return clause_match_by_cosine(embedded_chunks, embedded_references)
|