Add hypothesis template
This commit is contained in:
parent
d78fdb9c2e
commit
3d93d57820
|
@ -29,7 +29,8 @@ classifier = pipeline("zero-shot-classification",
|
||||||
|
|
||||||
classifier(
|
classifier(
|
||||||
"El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo",
|
"El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo",
|
||||||
candidate_labels=["cultura", "sociedad", "economia", "salud", "deportes"]
|
candidate_labels=["cultura", "sociedad", "economia", "salud", "deportes"],
|
||||||
|
hypothesis_template="Este ejemplo es {}."
|
||||||
)
|
)
|
||||||
"""output
|
"""output
|
||||||
{'sequence': 'El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo',
|
{'sequence': 'El autor se perfila, a los 50 años de su muerte, como uno de los grandes de su siglo',
|
||||||
|
|
Loading…
Reference in New Issue