|
--- |
|
license: mit |
|
datasets: |
|
- Gustavosta/Stable-Diffusion-Prompts |
|
- bartman081523/stable-diffusion-discord-prompts |
|
- Ar4ikov/sd_filtered_2m |
|
language: |
|
- en |
|
library_name: transformers |
|
pipeline_tag: text-generation |
|
tags: |
|
- art |
|
- code |
|
--- |
|
|
|
# Stable Diffusion Prompt Generator |
|
|
|
TODO: Complete me next time |
|
|
|
## Introcude |
|
|
|
... |
|
|
|
```python |
|
from transformers import pipeline |
|
|
|
pipe = pipeline('text-generation', model_id='Ar4ikov/gpt2-650k-stable-diffusion-prompt-generator') |
|
|
|
|
|
def get_valid_prompt(text: str) -> str: |
|
dot_split = text.split('.')[0] |
|
n_split = text.split('\n')[0] |
|
|
|
return { |
|
len(dot_split) < len(n_split): dot_split, |
|
len(n_split) > len(dot_split): n_split, |
|
len(n_split) == len(dot_split): dot_split |
|
}[True] |
|
|
|
|
|
prompt = 'A Tokio town landscape, sunset, by' |
|
|
|
valid_prompt = get_valid_prompt(pipe(prompt, max_length=77)[0]['generated_text']) |
|
print(valid_prompt) |
|
# >>> A Tokio town landscape, sunset, by Greg Rutkowski,Artgerm,trending on Behance,light effect,high detail,3d sculpture,golden ratio,dramatic,dramatic background,digital art |
|
``` |
