Tbruand commited on
Commit
8c88b4b
·
1 Parent(s): 0005ea7

feat(handler): affiche les scores de tous les labels formatés en markdown

Browse files
Files changed (1) hide show
  1. app/handler.py +5 -1
app/handler.py CHANGED
@@ -3,4 +3,8 @@ from models.zero_shot import ZeroShotModel
3
  model = ZeroShotModel()
4
 
5
  def predict(text: str) -> str:
6
- return model.predict(text)
 
 
 
 
 
3
  model = ZeroShotModel()
4
 
5
  def predict(text: str) -> str:
6
+ results = model.predict(text)
7
+ output = "### Résultat de la classification :\n\n"
8
+ for label, score in results:
9
+ output += f"- **{label}** : {score*100:.1f}%\n"
10
+ return output