--- base_model: google-t5/t5-base datasets: - gokaygokay/prompt-enhancer-dataset language: - en library_name: transformers license: apache-2.0 pipeline_tag: text2text-generation --- ```python from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM device = "cuda" if torch.cuda.is_available() else "cpu" # 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.2, device=device) max_target_length = 256 prefix = "enhance prompt: " short_prompt = "beautiful house with text 'hello'" 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 trim, large windows on the second floor, # three chimneys on the roof, green trees and shrubs in front of the house, # stone pathway leading to the front door, text on the house reads "hello" in all caps, # blue sky above, shadows cast by the trees, sunlight creating contrast on the house's facade, # some plants visible near the bottom right corner, overall warm and serene atmosphere. ```
