Views
No views yet
1from transformers import pipeline
2
3# Load the fine-tuned model
4generator = pipeline('text-generation', model='USACCAI/USACCAIV1')
5
6# Define a user story prompt
7prompt = "As a claims processor, when I check if a treatment occurred after the person has died, I need to show a message."
8
9# Generate acceptance criteria
10result = generator(prompt, max_length=100)
11
12# Print the generated output
13print(result)