Update app.py
Browse files
app.py
CHANGED
@@ -24,13 +24,13 @@ def ask(input_im, scale, steps, seed, images):
|
|
24 |
return images
|
25 |
|
26 |
def main(input_im, n_pairs, scale, steps, seed):
|
27 |
-
|
28 |
images = []
|
29 |
for i in range(n_pairs):
|
30 |
print('Asking for a new pair of image [' + str(i + 1) + '/' + str(n_pairs) + ']')
|
31 |
seed = seed+i
|
32 |
images = ask(input_im, scale, steps, seed, images)
|
33 |
-
|
34 |
return images
|
35 |
|
36 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
24 |
return images
|
25 |
|
26 |
def main(input_im, n_pairs, scale, steps, seed):
|
27 |
+
print('Start the magic !')
|
28 |
images = []
|
29 |
for i in range(n_pairs):
|
30 |
print('Asking for a new pair of image [' + str(i + 1) + '/' + str(n_pairs) + ']')
|
31 |
seed = seed+i
|
32 |
images = ask(input_im, scale, steps, seed, images)
|
33 |
+
print('Thanks to Sylvain, it worked like a charm!')
|
34 |
return images
|
35 |
|
36 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|