Spaces:
Running
Running
Commit
•
a6f67a5
1
Parent(s):
84aa40a
new model added
Browse files- app.py +1 -2
- vit-t-mae-pretrain.pt +2 -2
app.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
import torch
|
| 4 |
-
import torchvision
|
| 5 |
from PIL import Image
|
| 6 |
import numpy as np
|
| 7 |
import random
|
|
@@ -20,7 +19,7 @@ device = torch.device("cpu")
|
|
| 20 |
model.to(device)
|
| 21 |
|
| 22 |
transform = v2.Compose([
|
| 23 |
-
v2.Resize((
|
| 24 |
v2.ToTensor(),
|
| 25 |
v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
|
| 26 |
])
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
import torch
|
|
|
|
| 4 |
from PIL import Image
|
| 5 |
import numpy as np
|
| 6 |
import random
|
|
|
|
| 19 |
model.to(device)
|
| 20 |
|
| 21 |
transform = v2.Compose([
|
| 22 |
+
v2.Resize((96, 96)),
|
| 23 |
v2.ToTensor(),
|
| 24 |
v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
|
| 25 |
])
|
vit-t-mae-pretrain.pt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0abe9d41d86e1dab46b8ae0139e6bfd5c3e909c41ccb7b7d7e77f8558e4f6db1
|
| 3 |
+
size 29121704
|