Alexander CARNET commited on
Commit
a90d223
·
1 Parent(s): e97c50d

reparation du script python

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ import uvicorn
5
 
6
  app = FastAPI()
7
 
8
- class TextRequest:
9
  text : str
10
 
11
  #chargement de du modele ia SBERT
@@ -13,6 +13,6 @@ modele = SentenceTransformer('dangvantuan/sentence-camembert-base')
13
 
14
  @app.post("/vecteur")
15
  def creat_vector(data: TextRequest):
16
- vecteur = modele.encode(data.text).toList()
17
- print("vecteur send : " + vecteur)
18
  return {"vecteur":vecteur}
 
5
 
6
  app = FastAPI()
7
 
8
+ class TextRequest(BaseModel):
9
  text : str
10
 
11
  #chargement de du modele ia SBERT
 
13
 
14
  @app.post("/vecteur")
15
  def creat_vector(data: TextRequest):
16
+ vecteur = modele.encode(data.text).tolist()
17
+ print(f"Taille vecteur envoyé : {len(vecteur)}")
18
  return {"vecteur":vecteur}