Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ device = torch.device("cpu")
|
|
20 |
dinov2_vits14.to(device)
|
21 |
|
22 |
# Define the transformations: convert to tensor, resize, and normalize
|
23 |
-
transform_image =
|
24 |
|
25 |
model = torch.load("dress_model.pth")
|
26 |
model.eval()
|
|
|
20 |
dinov2_vits14.to(device)
|
21 |
|
22 |
# Define the transformations: convert to tensor, resize, and normalize
|
23 |
+
transform_image = transforms.Compose([transforms.ToTensor(), transforms.Resize(224), transforms.Normalize([0.5], [0.5])])
|
24 |
|
25 |
model = torch.load("dress_model.pth")
|
26 |
model.eval()
|