Commit
•
42e3bf6
1
Parent(s):
4e620f6
app.py fix
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ transform = T.Compose([
|
|
| 19 |
T.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),
|
| 20 |
])
|
| 21 |
|
| 22 |
-
model = torch.jit.load('
|
| 23 |
|
| 24 |
def inference(im):
|
| 25 |
img = transform(im).unsqueeze(0)
|
|
|
|
| 19 |
T.Normalize(IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD),
|
| 20 |
])
|
| 21 |
|
| 22 |
+
model = torch.jit.load('686_convnext_base_siamfly_backbone_jit.pt')
|
| 23 |
|
| 24 |
def inference(im):
|
| 25 |
img = transform(im).unsqueeze(0)
|