Spaces:
Sleeping
Sleeping
small fix
Browse files
app.py
CHANGED
|
@@ -102,7 +102,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
|
|
| 102 |
input_image = input_image.resize((size, int(size * input_image.height / input_image.width)))
|
| 103 |
else:
|
| 104 |
input_image = input_image.resize((int(size * input_image.width / input_image.height), size))
|
| 105 |
-
tensor = TF.to_tensor(
|
| 106 |
|
| 107 |
original_size = tensor.shape[-2:]
|
| 108 |
padded_tensor, padding = pad_for_model(tensor, args.round_multiple)
|
|
|
|
| 102 |
input_image = input_image.resize((size, int(size * input_image.height / input_image.width)))
|
| 103 |
else:
|
| 104 |
input_image = input_image.resize((int(size * input_image.width / input_image.height), size))
|
| 105 |
+
tensor = TF.to_tensor(input_image)
|
| 106 |
|
| 107 |
original_size = tensor.shape[-2:]
|
| 108 |
padded_tensor, padding = pad_for_model(tensor, args.round_multiple)
|