Hatman commited on
Commit
810bc7a
·
verified ·
1 Parent(s): f37caca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.LANCZOS)
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)