Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def prepare_image(image_path_or_url):
|
|
| 22 |
# Convert to tensor and move to correct device and dtype
|
| 23 |
transform = transforms.Compose([
|
| 24 |
transforms.ToTensor(),
|
| 25 |
-
transforms.Resize((1024, 1024), interpolation=transforms.InterpolationMode.
|
| 26 |
])
|
| 27 |
image_tensor = transform(image).unsqueeze(0) # Add batch dimension
|
| 28 |
return image_tensor.to(device=device, dtype=dtype)
|
|
|
|
| 22 |
# Convert to tensor and move to correct device and dtype
|
| 23 |
transform = transforms.Compose([
|
| 24 |
transforms.ToTensor(),
|
| 25 |
+
transforms.Resize((1024, 1024), interpolation=transforms.InterpolationMode.BICUBIC)
|
| 26 |
])
|
| 27 |
image_tensor = transform(image).unsqueeze(0) # Add batch dimension
|
| 28 |
return image_tensor.to(device=device, dtype=dtype)
|