Named entities: | |
""" | |
sequences = pipe( | |
prompt, | |
max_new_tokens=15, | |
return_full_text = False, | |
) | |
for seq in sequences: | |
print(f"{seq['generated_text']}") | |
- Golden State Warriors | |
- San Francisco | |
As you can see, the model correctly identified two named entities from the given text. |