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"
|