wwnvp01 commited on
Commit
cf741e2
·
verified ·
1 Parent(s): d37c0df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = T.Compose([T.ToTensor(), T.Resize(224), T.Normalize([0.5], [0.5])])
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()