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