Spaces:
Sleeping
Sleeping
Update model_loader.py
Browse files- model_loader.py +0 -2
model_loader.py
CHANGED
|
@@ -4,8 +4,6 @@ from torchvision import models, transforms
|
|
| 4 |
from PIL import Image
|
| 5 |
import os
|
| 6 |
|
| 7 |
-
DEFAULT_IMG_SIZE = int(os.getenv("IMG_SIZE", "32"))
|
| 8 |
-
|
| 9 |
def build_alexnet(num_classes=2):
|
| 10 |
model = models.alexnet(pretrained=False)
|
| 11 |
in_features = model.classifier[6].in_features
|
|
|
|
| 4 |
from PIL import Image
|
| 5 |
import os
|
| 6 |
|
|
|
|
|
|
|
| 7 |
def build_alexnet(num_classes=2):
|
| 8 |
model = models.alexnet(pretrained=False)
|
| 9 |
in_features = model.classifier[6].in_features
|