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