from transformers import pipeline, set_seed
path = "akahana/harry-potter-gpt2"
generator = pipeline('text-generation',
model=path)
set_seed(42)
kalimat = "sec"
preds = generator(kalimat,
max_length=25,
num_return_sequences=1)
for data in preds:
print(data)
- Downloads last month
- 3