cdnuts commited on
Commit
ae2d2c2
1 Parent(s): bf99018

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ snapshot_download(repo_id="Thouph/eva-vit-1b-224-8043", local_dir = "./")
11
  model = torch.load('model.pth', map_location=torch.device('cpu'))
12
  model.eval()
13
  transform = transforms.Compose([
14
- transforms.Resize((448, 448)),
15
  transforms.ToTensor(),
16
  transforms.Normalize(mean=[
17
  0.48145466,
 
11
  model = torch.load('model.pth', map_location=torch.device('cpu'))
12
  model.eval()
13
  transform = transforms.Compose([
14
+ transforms.Resize((224, 224)),
15
  transforms.ToTensor(),
16
  transforms.Normalize(mean=[
17
  0.48145466,