Views
No views yet
1from transformers import pipeline
2
3generator = pipeline("text-generation", model = "KennethTM/gpt2-small-danish-review-response")
4
5def prompt_template(user, review):
6 return f"### Bruger:\n{user}\n\n### Anmeldelse:\n{review}\n\n### Svar:\nKære {user}\n"
7
8prompt = prompt_template(user = "Anders", review = "Umuligt at komme igennem på telefonen.")
9
10text = generator(prompt)
11
12print(text[0]["generated_text"])1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("KennethTM/gpt2-small-danish-review-response")
4model = AutoModelForCausalLM.from_pretrained("KennethTM/gpt2-small-danish-review-response")