Update script.py

Annotate code:
# num_features = model.get_classifier().in_features
# model.classifier = nn.Linear(num_features, number_of_categories)

script.py

@@ -36,8 +36,8 @@ class PytorchWorker:
             model.load_state_dict(model_ckpt, strict=False)
             msg = model.load_state_dict(model_ckpt, strict=False)
             print("load_state_dict: ", msg)
-            num_features = model.get_classifier().in_features
-            model.classifier = nn.Linear(num_features, number_of_categories)
+            # num_features = model.get_classifier().in_features
+            # model.classifier = nn.Linear(num_features, number_of_categories)
 
             return model.to(self.device).eval()