Boboiazumi commited on
Commit
a6e70ac
·
verified ·
1 Parent(s): 766cd7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.3),
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")