Dehnes's picture
Update README.md
6d8ea6d
|
raw
history blame
768 Bytes

pipeline_tag: zero-shot-classification

widget:

  • text: "Letzte Woche gab es einen Selbstmord in einer nahe gelegenen kolonie"

    candidate_labels: "Verbrechen,Tragödie,Stehlen"

    hypothesis_template: "In deisem geht es um {}."


from transformers import pipeline

classifier = pipeline("zero-shot-classification",

                      model="Dehnes/zeroshot_gbert")

sequence = "Letzte Woche gab es einen Selbstmord in einer nahe gelegenen Stadt"

candidate_labels = ["Verbrechen","Tragödie","Stehlen"]

hypothesis_template = "In diesem Satz geht es um das Thema {}."     ## Since monolingual model,its sensitive to hypothesis template. This can be experimented

classifier(sequence, candidate_labels, hypothesis_template=hypothesis_template)