new datasets fix

models.py

@@ -37,7 +37,7 @@ def load_model3(weights_path='weights/class_2.pth'):
         model3 = models.mobilenet_v2(pretrained=False)
         for param in model3.features.parameters():
             param.requires_grad = False
-        model3.classifier[1] = nn.Linear(model3.classifier[1].in_features, 6)
+        model3.classifier[1] = nn.Linear(model3.classifier[1].in_features, 2)
         state_dict = torch.load(weights_path, map_location=DEVICE)
         model3.load_state_dict(state_dict)
         model3.eval()

weights/class.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0720a4afc5d0649b9af6e1532947af363f7230276172f6345aae6b951be071e5
-size 9205515

weights/class2.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bf1835ce202339850361b67f914d7768400f981aa2f0c1a6a29e7f268a749f18
-size 9174411