Spaces:
Runtime error
Runtime error
fake fix
Browse files
app.py
CHANGED
|
@@ -5,6 +5,7 @@ from torchvision import transforms
|
|
| 5 |
|
| 6 |
model = torch.hub.load('nicolalandro/ntsnet-cub200', 'ntsnet', pretrained=True,
|
| 7 |
**{'topN': 6, 'device': 'cpu', 'num_classes': 200}).eval()
|
|
|
|
| 8 |
transform_test = transforms.Compose([
|
| 9 |
transforms.Resize((600, 600), Image.BILINEAR),
|
| 10 |
transforms.CenterCrop((448, 448)),
|
|
|
|
| 5 |
|
| 6 |
model = torch.hub.load('nicolalandro/ntsnet-cub200', 'ntsnet', pretrained=True,
|
| 7 |
**{'topN': 6, 'device': 'cpu', 'num_classes': 200}).eval()
|
| 8 |
+
|
| 9 |
transform_test = transforms.Compose([
|
| 10 |
transforms.Resize((600, 600), Image.BILINEAR),
|
| 11 |
transforms.CenterCrop((448, 448)),
|