Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ transform = transforms.Compose([
|
|
13 |
|
14 |
model = models.mobilenet_v3_large(pretrained=True)
|
15 |
model.classifier[3] = nn.Sequential(
|
16 |
-
nn.Dropout(0.
|
17 |
nn.Linear(model.classifier[3].in_features, 2)
|
18 |
)
|
19 |
model = model.to("cpu")
|
|
|
13 |
|
14 |
model = models.mobilenet_v3_large(pretrained=True)
|
15 |
model.classifier[3] = nn.Sequential(
|
16 |
+
nn.Dropout(0.2),
|
17 |
nn.Linear(model.classifier[3].in_features, 2)
|
18 |
)
|
19 |
model = model.to("cpu")
|