Update app.py
Browse files
app.py
CHANGED
|
@@ -43,10 +43,10 @@ footer {
|
|
| 43 |
'''
|
| 44 |
|
| 45 |
# Functions
|
| 46 |
-
def save_image(img):
|
| 47 |
-
|
| 48 |
-
img.save(
|
| 49 |
-
return
|
| 50 |
|
| 51 |
def get_seed(seed):
|
| 52 |
seed = seed.strip()
|
|
@@ -71,14 +71,14 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
|
|
| 71 |
"num_inference_steps": steps,
|
| 72 |
"guidance_scale": guidance,
|
| 73 |
"num_images_per_prompt": number,
|
| 74 |
-
"cross_attention_kwargs": {"scale": 0.
|
| 75 |
"generator": torch.Generator().manual_seed(seed),
|
| 76 |
"use_resolution_binning": True,
|
| 77 |
"output_type":"pil",
|
| 78 |
}
|
| 79 |
|
| 80 |
images = model(**parameters).images
|
| 81 |
-
image_paths = [save_image(img) for img in images]
|
| 82 |
print(image_paths)
|
| 83 |
return image_paths
|
| 84 |
|
|
|
|
| 43 |
'''
|
| 44 |
|
| 45 |
# Functions
|
| 46 |
+
def save_image(img, seed):
|
| 47 |
+
name = f"{seed}-{uuid.uuid4()}.png"
|
| 48 |
+
img.save(name)
|
| 49 |
+
return name
|
| 50 |
|
| 51 |
def get_seed(seed):
|
| 52 |
seed = seed.strip()
|
|
|
|
| 71 |
"num_inference_steps": steps,
|
| 72 |
"guidance_scale": guidance,
|
| 73 |
"num_images_per_prompt": number,
|
| 74 |
+
"cross_attention_kwargs": {"scale": 0.01},
|
| 75 |
"generator": torch.Generator().manual_seed(seed),
|
| 76 |
"use_resolution_binning": True,
|
| 77 |
"output_type":"pil",
|
| 78 |
}
|
| 79 |
|
| 80 |
images = model(**parameters).images
|
| 81 |
+
image_paths = [save_image(img, seed) for img in images]
|
| 82 |
print(image_paths)
|
| 83 |
return image_paths
|
| 84 |
|