antonovmaxim commited on
Commit
dc02fd8
·
1 Parent(s): a523faa

Update app.py

Browse files

Delete augmentation function

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model.to(device)
15
  model.eval()
16
 
17
  transform = transforms.Compose([
18
- transforms.RandomHorizontalFlip(p=0.5),
19
  transforms.Resize(256), # Resize the image to 256x256 pixels
20
  transforms.CenterCrop(224), # Crop the center 224x224 pixels
21
  transforms.ToTensor(), # Convert the image to a PyTorch tensor
 
15
  model.eval()
16
 
17
  transform = transforms.Compose([
18
+ # transforms.RandomHorizontalFlip(p=0.5),
19
  transforms.Resize(256), # Resize the image to 256x256 pixels
20
  transforms.CenterCrop(224), # Crop the center 224x224 pixels
21
  transforms.ToTensor(), # Convert the image to a PyTorch tensor