Alexander CARNET commited on
Commit
15268e6
·
1 Parent(s): 76ff603

ajout du script ia SBERT dans la route /vecteur

Browse files
Files changed (2) hide show
  1. app.py +11 -3
  2. requirements.txt +3 -1
app.py CHANGED
@@ -1,7 +1,15 @@
1
  from fastapi import FastAPI
 
 
 
2
 
3
  app = FastAPI()
4
 
5
- @app.get("/")
6
- def greet_json():
7
- return {"Hello": "World!"}
 
 
 
 
 
 
1
  from fastapi import FastAPI
2
+ from sentence_transformers import SentenceTransformer
3
+ from pydantic import BaseModel
4
+ import uvicorn
5
 
6
  app = FastAPI()
7
 
8
+ //chargement de du modele ia SBERT
9
+ modele = SentenceTransformer('dangvantuan/sentence-camembert-base')
10
+
11
+ @app.post("/vecteur")
12
+ def creat_vector(data: TextRequest):
13
+ vecteur = modele.encode(data.text).toList()
14
+ print("vecteur send : " + vecteur)
15
+ return {"vecteur":vecteur}
requirements.txt CHANGED
@@ -1,2 +1,4 @@
1
  fastapi
2
- uvicorn[standard]
 
 
 
1
  fastapi
2
+ uvicorn
3
+ sentence-transformers
4
+ pydantic