Update app.py
Browse files
app.py
CHANGED
|
@@ -581,7 +581,7 @@ def test_single_image(generator_path, input_image_path):
|
|
| 581 |
generator.eval()
|
| 582 |
|
| 583 |
# Preprocess the input image
|
| 584 |
-
input_image = preprocess_input(input_image_path).to(device)
|
| 585 |
|
| 586 |
# Perform inference
|
| 587 |
with torch.no_grad():
|
|
|
|
| 581 |
generator.eval()
|
| 582 |
|
| 583 |
# Preprocess the input image
|
| 584 |
+
input_image = preprocess_input(input_image_path).to(device)
|
| 585 |
|
| 586 |
# Perform inference
|
| 587 |
with torch.no_grad():
|