Spaces:
Running
on
Zero
Running
on
Zero
no image degradation, swapped output images
Browse filesnow output image is saved with the LoRA in PNG format
app.py
CHANGED
|
@@ -56,7 +56,7 @@ def generate_image(prompt, height, width, num_inference_steps, seed=42, randomiz
|
|
| 56 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 57 |
).images[0]
|
| 58 |
|
| 59 |
-
return (image_no_lora
|
| 60 |
|
| 61 |
|
| 62 |
# Example prompts
|
|
@@ -135,8 +135,9 @@ with gr.Blocks() as demo:
|
|
| 135 |
)
|
| 136 |
with gr.Column(scale=1):
|
| 137 |
output_image = gr.ImageSlider(
|
| 138 |
-
label="Output (Left-
|
| 139 |
type="pil",
|
|
|
|
| 140 |
)
|
| 141 |
|
| 142 |
|
|
|
|
| 56 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
| 57 |
).images[0]
|
| 58 |
|
| 59 |
+
return (image, image_no_lora), seed
|
| 60 |
|
| 61 |
|
| 62 |
# Example prompts
|
|
|
|
| 135 |
)
|
| 136 |
with gr.Column(scale=1):
|
| 137 |
output_image = gr.ImageSlider(
|
| 138 |
+
label="Output (Left-without the LoRA, Right-with)",
|
| 139 |
type="pil",
|
| 140 |
+
format="png"
|
| 141 |
)
|
| 142 |
|
| 143 |
|