Spaces:
Runtime error
Runtime error
antonovmaxim
commited on
Commit
·
dc02fd8
1
Parent(s):
a523faa
Update app.py
Browse filesDelete augmentation function
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
|