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")
|