Update unified_cell_classifier.py

unified_cell_classifier.py

@@ -252,7 +252,7 @@ class UnifiedCellClassifier(nn.Module):
         # Carica i pesi del main classifier
         main_weights_path = get_file_path("main_classifier/main_classifier.bin")
         main_state_dict = torch.load(main_weights_path)
-        model.main_classifier.load_state_dict(main_state_dict, , strict=False)
+        model.main_classifier.load_state_dict(main_state_dict, strict=False)
         
         # Carica le label del main classifier
         main_labels_path = get_file_path("main_classifier/id2label_main.json")
@@ -272,7 +272,7 @@ class UnifiedCellClassifier(nn.Module):
                 # Carica i pesi del sub-classificatore
                 sub_weights_path = get_file_path(f"sub_classifiers/{sub_name}.bin")
                 sub_state_dict = torch.load(sub_weights_path) 
-                model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, , strict=False)
+                model.sub_classifiers[sub_name].load_state_dict(sub_state_dict, strict=False)
                 
                 # Carica le label del sub-classificatore
                 sub_labels_path = get_file_path(f"sub_classifiers/{sub_name}_id2label.json")