Views
No views yet
1from transformers import pipeline
2
3pipe = pipeline("text-generation", model="nroggendorff/vegetarian-mayo")
4
5question = "What color is the sky?"
6conv = [{"role": "user", "content": question}]
7
8response = pipe(conv, max_new_tokens=32)[0]['generated_text'][-1]['content']
9print(response)