Spaces:
Running
on
Zero
Running
on
Zero
change metadata label to parameters
Browse files
app.py
CHANGED
@@ -107,7 +107,7 @@ def generate(
|
|
107 |
|
108 |
if use_upscaler:
|
109 |
upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
|
110 |
-
|
111 |
"prompt": prompt,
|
112 |
"negative_prompt": negative_prompt,
|
113 |
"resolution": f"{width} x {height}",
|
@@ -118,23 +118,25 @@ def generate(
|
|
118 |
"sdxl_style": style_selector,
|
119 |
"add_quality_tags": add_quality_tags,
|
120 |
"quality_tags": quality_selector,
|
121 |
-
"Model hash": "e3c47aedb0",
|
122 |
-
"Model": "animagine-xl-3.1",
|
123 |
}
|
124 |
|
125 |
if use_upscaler:
|
126 |
new_width = int(width * upscale_by)
|
127 |
new_height = int(height * upscale_by)
|
128 |
-
|
129 |
"upscale_method": "nearest-exact",
|
130 |
"upscaler_strength": upscaler_strength,
|
131 |
"upscale_by": upscale_by,
|
132 |
"new_resolution": f"{new_width} x {new_height}",
|
133 |
}
|
134 |
else:
|
135 |
-
|
136 |
-
|
137 |
-
|
|
|
|
|
|
|
|
|
138 |
|
139 |
try:
|
140 |
if use_upscaler:
|
@@ -173,14 +175,14 @@ def generate(
|
|
173 |
|
174 |
if images:
|
175 |
image_paths = [
|
176 |
-
utils.save_image(image,
|
177 |
for image in images
|
178 |
]
|
179 |
|
180 |
for image_path in image_paths:
|
181 |
logger.info(f"Image saved as {image_path} with metadata")
|
182 |
|
183 |
-
return image_paths,
|
184 |
except Exception as e:
|
185 |
logger.exception(f"An error occurred: {e}")
|
186 |
raise
|
|
|
107 |
|
108 |
if use_upscaler:
|
109 |
upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
|
110 |
+
parameters = {
|
111 |
"prompt": prompt,
|
112 |
"negative_prompt": negative_prompt,
|
113 |
"resolution": f"{width} x {height}",
|
|
|
118 |
"sdxl_style": style_selector,
|
119 |
"add_quality_tags": add_quality_tags,
|
120 |
"quality_tags": quality_selector,
|
|
|
|
|
121 |
}
|
122 |
|
123 |
if use_upscaler:
|
124 |
new_width = int(width * upscale_by)
|
125 |
new_height = int(height * upscale_by)
|
126 |
+
parameters["use_upscaler"] = {
|
127 |
"upscale_method": "nearest-exact",
|
128 |
"upscaler_strength": upscaler_strength,
|
129 |
"upscale_by": upscale_by,
|
130 |
"new_resolution": f"{new_width} x {new_height}",
|
131 |
}
|
132 |
else:
|
133 |
+
parameters["use_upscaler"] = None
|
134 |
+
parameters["Model"] = {
|
135 |
+
"Model": DESCRIPTION,
|
136 |
+
"Model hash": "e3c47aedb0",
|
137 |
+
}
|
138 |
+
|
139 |
+
logger.info(json.dumps(parameters, indent=4))
|
140 |
|
141 |
try:
|
142 |
if use_upscaler:
|
|
|
175 |
|
176 |
if images:
|
177 |
image_paths = [
|
178 |
+
utils.save_image(image, parameters, OUTPUT_DIR, IS_COLAB)
|
179 |
for image in images
|
180 |
]
|
181 |
|
182 |
for image_path in image_paths:
|
183 |
logger.info(f"Image saved as {image_path} with metadata")
|
184 |
|
185 |
+
return image_paths, parameters
|
186 |
except Exception as e:
|
187 |
logger.exception(f"An error occurred: {e}")
|
188 |
raise
|