Spaces:
Runtime error
Runtime error
ChenyangSi
commited on
Commit
•
693cf66
1
Parent(s):
7f15b2e
Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
|
|
49 |
|
50 |
torch.manual_seed(seed)
|
51 |
print("Generating SD:")
|
52 |
-
sd_image = pip(prompt
|
53 |
sd_image_prev = sd_image
|
54 |
else:
|
55 |
sd_image = sd_image_prev
|
@@ -60,7 +60,7 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
|
|
60 |
|
61 |
torch.manual_seed(seed)
|
62 |
print("Generating FreeU:")
|
63 |
-
freeu_image = pip(prompt
|
64 |
|
65 |
# First SD, then freeu
|
66 |
images = [sd_image, freeu_image]
|
|
|
49 |
|
50 |
torch.manual_seed(seed)
|
51 |
print("Generating SD:")
|
52 |
+
sd_image = pip(prompt).images[0]
|
53 |
sd_image_prev = sd_image
|
54 |
else:
|
55 |
sd_image = sd_image_prev
|
|
|
60 |
|
61 |
torch.manual_seed(seed)
|
62 |
print("Generating FreeU:")
|
63 |
+
freeu_image = pip(prompt).images[0]
|
64 |
|
65 |
# First SD, then freeu
|
66 |
images = [sd_image, freeu_image]
|