Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -130,7 +130,7 @@ def stylize_image(content_image: Image.Image):
|
|
| 130 |
generated_img = new_content_img.clone().requires_grad_(True).to(device)
|
| 131 |
|
| 132 |
# 3. Setup optimizer for the generated image
|
| 133 |
-
lr = 0.
|
| 134 |
optimizer = optim.Adam([generated_img], lr=lr)
|
| 135 |
|
| 136 |
# 4. Run optimization loop
|
|
|
|
| 130 |
generated_img = new_content_img.clone().requires_grad_(True).to(device)
|
| 131 |
|
| 132 |
# 3. Setup optimizer for the generated image
|
| 133 |
+
lr = 0.002
|
| 134 |
optimizer = optim.Adam([generated_img], lr=lr)
|
| 135 |
|
| 136 |
# 4. Run optimization loop
|