Spaces:
Running
on
T4
Running
on
T4
Krebzonide
commited on
Commit
·
b21649c
1
Parent(s):
b301caf
Update app.py
Browse files
app.py
CHANGED
@@ -25,11 +25,12 @@ def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tq
|
|
25 |
guidance_scale=0.0,
|
26 |
generator=torch.manual_seed(seed),
|
27 |
).images
|
28 |
-
upscaled_images =
|
29 |
final_images = img2img(
|
30 |
prompt,
|
31 |
num_inference_steps=samp_steps,
|
32 |
guidance_scale=5,
|
|
|
33 |
generator=torch.manual_seed(seed),
|
34 |
).images
|
35 |
return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(final_images)]), seed
|
|
|
25 |
guidance_scale=0.0,
|
26 |
generator=torch.manual_seed(seed),
|
27 |
).images
|
28 |
+
upscaled_images = img.resize((1024,1024), 1) for img in images
|
29 |
final_images = img2img(
|
30 |
prompt,
|
31 |
num_inference_steps=samp_steps,
|
32 |
guidance_scale=5,
|
33 |
+
strength=0.5,
|
34 |
generator=torch.manual_seed(seed),
|
35 |
).images
|
36 |
return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(final_images)]), seed
|