Spaces:
Sleeping
Sleeping
Alexander CARNET commited on
Commit ·
a90d223
1
Parent(s): e97c50d
reparation du script python
Browse files
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).
|
| 17 |
-
print("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}
|