File size: 1,217 Bytes
5b465f6 4b8c3b6 5b465f6 5309247 2fd5ae5 5b465f6 53bbd9c 4b8c3b6 5b465f6 4b8c3b6 5b465f6 4b8c3b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
---
library_name: transformers
license: apache-2.0
datasets:
- gokaygokay/prompt-enhancer-dataset
language:
- en
base_model: google-t5/t5-base
pipeline_tag: text2text-generation
---
```python
from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
# Model checkpoint
model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
# Tokenizer
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
# Model
model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
enhancer = pipeline('text2text-generation', model=model, tokenizer=tokenizer, repetition_penalty=1.5, device="cuda")
max_target_length = 256
prefix = "enhance prompt: "
short_prompt = "beautiful house"
answer = enhancer(prefix + short_prompt, max_length=max_target_length)
final_answer = answer[0]['generated_text']
print(final_answer)
# a two-story house with white walls and dark brown roof, large windows on the upper floor, black shutters on the windows, a chimney on top of one of the windows, a small balcony with a metal railing in front of the house, a tree with green leaves to its left side, another tree with green foliage to their right, clear blue sky above, sunlight casting shadows from the trees onto the house.
``` |