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,
|