Alexander CARNET commited on
Commit
c9f0096
·
1 Parent(s): 9a9a1ed

modification et baisse de la version de transforme

Browse files
Files changed (2) hide show
  1. app.py +6 -11
  2. requirements.txt +2 -1
app.py CHANGED
@@ -18,18 +18,13 @@ class TextRequestAvecEtiquettes(BaseModel):
18
 
19
  # Chargement de SBERT
20
  print("Chargement de SBERT...")
21
- modele_sbert = SentenceTransformer('alex246879/sbert-professionnel-2026')
 
22
  print("SBERT chargé avec succès")
23
 
24
- modele_gliner = None
25
-
26
- def get_gliner():
27
- global modele_gliner
28
- if modele_gliner is None:
29
- print("Chargement de GLiNER...")
30
- modele_gliner = GLiNER.from_pretrained("urchade/gliner_multi_pii-v1")
31
- print("GLiNER chargé avec succès")
32
- return modele_gliner
33
 
34
  @app.post("/vecteur")
35
  def creat_vector(data: TextRequest):
@@ -45,7 +40,7 @@ def creat_vectors(data: TextRequestAvecEtiquettes):
45
 
46
  # Extraction des entités par étiquette via GLiNER
47
  print("Extraction des entités via GLiNER...")
48
- entites = get_gliner().predict_entities(data.text, data.etiquettes)
49
  print(f"Entités extraites : {entites}")
50
 
51
  # Regroupement des textes extraits par étiquette
 
18
 
19
  # Chargement de SBERT
20
  print("Chargement de SBERT...")
21
+ modele_sbert = SentenceTransformer('paraphrase-multilingual-mpnet-base-v2')
22
+ #modele_sbert = SentenceTransformer('alex246879/sbert-professionnel-2026')
23
  print("SBERT chargé avec succès")
24
 
25
+ print("Chargement de GLiNER...")
26
+ modele_gliner = GLiNER.from_pretrained("urchade/gliner_multi_pii-v1")
27
+ print("GLiNER chargé avec succès")
 
 
 
 
 
 
28
 
29
  @app.post("/vecteur")
30
  def creat_vector(data: TextRequest):
 
40
 
41
  # Extraction des entités par étiquette via GLiNER
42
  print("Extraction des entités via GLiNER...")
43
+ entites = modele_gliner.predict_entities(data.text, data.etiquettes)
44
  print(f"Entités extraites : {entites}")
45
 
46
  # Regroupement des textes extraits par étiquette
requirements.txt CHANGED
@@ -2,4 +2,5 @@ fastapi
2
  uvicorn
3
  pydantic
4
  gliner
5
- sentence-transformers
 
 
2
  uvicorn
3
  pydantic
4
  gliner
5
+ sentence-transformers
6
+ transformers==4.43.4