File size: 296 Bytes
9decc9d 3ece550 9decc9d 3ece550 8c88b4b |
1 2 3 4 5 6 7 8 9 10 |
from models.zero_shot import ZeroShotModel
model = ZeroShotModel()
def predict(text: str) -> str:
results = model.predict(text)
output = "### Résultat de la classification :\n\n"
for label, score in results:
output += f"- **{label}** : {score*100:.1f}%\n"
return output |