Davidzhangyuanhan commited on
Commit
43382ea
1 Parent(s): 811c881

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def pil_loader(filepath):
22
 
23
  def build_transforms(input_size, center_crop=True):
24
  transform = torchvision.transforms.Compose([
25
- torchvision.transforms.ToPILImage()
26
  torchvision.transforms.Resize(input_size * 8 // 7),
27
  torchvision.transforms.CenterCrop(input_size),
28
  torchvision.transforms.ToTensor(),
 
22
 
23
  def build_transforms(input_size, center_crop=True):
24
  transform = torchvision.transforms.Compose([
25
+ torchvision.transforms.ToPILImage(),
26
  torchvision.transforms.Resize(input_size * 8 // 7),
27
  torchvision.transforms.CenterCrop(input_size),
28
  torchvision.transforms.ToTensor(),