Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load the model
5tokenizer = AutoTokenizer.from_pretrained("TurkuNLP/gpt3-finnish-small")
6base_model = AutoModelForCausalLM.from_pretrained("TurkuNLP/gpt3-finnish-small")
7model = PeftModel.from_pretrained(base_model, "swnometal666/finnish-healthcare-terminology-v4")
8
9# Generate healthcare-related text
10prompt = "Kysymys: Mitä on diabetes? Vastaus:"
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_new_tokens=50)
13response = tokenizer.decode(outputs[0], skip_special_tokens=True)
14print(response)