ALLOUNE commited on
Commit
555cb01
·
1 Parent(s): 0f73722

add text to heatmap

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -14,7 +14,7 @@ class Output(BaseModel):
14
  matrix : List
15
 
16
  @app.post("/process", response_model=Output)
17
- def process(payload: Input):
18
  saoke_spec = text_to_saoke(payload.text1)
19
  saoke_patent = text_to_saoke(payload.text2)
20
 
@@ -22,6 +22,13 @@ def process(payload: Input):
22
 
23
  embeddings1, embeddings2 = text_to_embeddings(saoke_spec, saoke_patent, model)
24
 
 
 
 
 
 
 
 
25
  matrix = embeddings_to_matrix(embeddings1, embeddings2)
26
  print({"matrix": matrix})
27
  return {"matrix": matrix}
 
14
  matrix : List
15
 
16
  @app.post("/process", response_model=Output)
17
+ def saoke_to_heatmap(payload: Input):
18
  saoke_spec = text_to_saoke(payload.text1)
19
  saoke_patent = text_to_saoke(payload.text2)
20
 
 
22
 
23
  embeddings1, embeddings2 = text_to_embeddings(saoke_spec, saoke_patent, model)
24
 
25
+ matrix = embeddings_to_matrix(embeddings1, embeddings2)
26
+ print({"matrix": matrix})
27
+ return {"matrix": matrix}
28
+
29
+ @app.post("/text-to-heatmap", response_model=Output)
30
+ def text_to_heatmap(payload: Input):
31
+ embeddings1, embeddings2 = text_to_embeddings(payload.text1, payload.text2)
32
  matrix = embeddings_to_matrix(embeddings1, embeddings2)
33
  print({"matrix": matrix})
34
  return {"matrix": matrix}