Spaces:

Ymokay
/

ToxiCheck

Sleeping

ToxiCheck / app /handler.py

Tbruand

feat(models): ajoute un modèle de classification zero-shot basé sur bart-large-mnli

9decc9d 2 months ago

130 Bytes

	from models.zero_shot import ZeroShotModel

	model = ZeroShotModel()

	def predict(text: str) -> str:
	return model.predict(text)