Spaces:
Running
on
T4
Running
on
T4
Krebzonide
commited on
Commit
·
a9201cc
1
Parent(s):
aabe0fe
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
|
|
33 |
strength=0.5,
|
34 |
generator=torch.manual_seed(seed),
|
35 |
).images[0]
|
36 |
-
return
|
37 |
|
38 |
def set_base_models():
|
39 |
txt2img = AutoPipelineForText2Image.from_pretrained(
|
|
|
33 |
strength=0.5,
|
34 |
generator=torch.manual_seed(seed),
|
35 |
).images[0]
|
36 |
+
return final_image, seed
|
37 |
|
38 |
def set_base_models():
|
39 |
txt2img = AutoPipelineForText2Image.from_pretrained(
|