Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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((
|
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,
|