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 |
|