Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ app = FastAPI()
|
|
| 22 |
@app.get("/")
|
| 23 |
|
| 24 |
def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, width=896, height=1152):
|
| 25 |
-
result = {"prompt":
|
| 26 |
print(result)
|
| 27 |
try:
|
| 28 |
language = detect(prompt)
|
|
@@ -34,6 +34,10 @@ def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cf
|
|
| 34 |
|
| 35 |
prompt = re.sub(r'[^a-zA-Zа-яА-Я\s]', '', prompt)
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
cfg = int(cfg_scale)
|
| 38 |
steps = int(steps)
|
| 39 |
seed = int(seed)
|
|
|
|
| 22 |
@app.get("/")
|
| 23 |
|
| 24 |
def query(prompt, negative_prompt, task, steps=35, sampler="DPM++ 2M Karras", cfg_scale=7, seed=-1, width=896, height=1152):
|
| 25 |
+
result = {prompt:"prompt", negative_prompt:"negative_prompt", task:"task", steps:"steps", sampler:"sampler", cfg_scale:"cfg_scale", seed:"seed" , width:"width" , height:"height" }
|
| 26 |
print(result)
|
| 27 |
try:
|
| 28 |
language = detect(prompt)
|
|
|
|
| 34 |
|
| 35 |
prompt = re.sub(r'[^a-zA-Zа-яА-Я\s]', '', prompt)
|
| 36 |
|
| 37 |
+
# Add some extra flair to the prompt
|
| 38 |
+
prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
|
| 39 |
+
print(f'\033[1mGeneration {key}:\033[0m {prompt}')
|
| 40 |
+
|
| 41 |
cfg = int(cfg_scale)
|
| 42 |
steps = int(steps)
|
| 43 |
seed = int(seed)
|