Refactor: update import paths and model type for scBloodClassifier; remove unified model implementation

class_registration.py

@@ -1,5 +1,5 @@
 # Import the custom classes
-from .modeling_unified import scBloodClassifierConfig, scBloodClassifier
+from .modeling_scBloodClassifier import scBloodClassifierConfig, scBloodClassifier
 
 # Import the necessary Auto classes from transformers
 from transformers import AutoConfig, AutoModel

config.json

@@ -32,7 +32,7 @@
     "7": "TRAV1-2- CD8+ T cells",
     "8": "gd T cells"
   },
-  "model_type": "unified-cell-classifier",
+  "model_type": "scBloodClassifier",
   "sub_classifier_names": [
     "B_cells_classifier",
     "CD4plus_T_cells_classifier",

modeling_unified.py → modeling_scBloodClassifier.py

@@ -5,6 +5,7 @@ import torch
 import torch.nn as nn
 from transformers import PretrainedConfig, PreTrainedModel
 from transformers.modeling_outputs import SequenceClassifierOutput
+from transformers import AutoConfig, AutoModel
 
 
 class MLPBlock(nn.Module):
@@ -202,3 +203,9 @@ class scBloodClassifier(PreTrainedModel):
         if not os.path.exists(readme_path):
             with open(readme_path, "w") as f:
                 f.write("# scBloodClassifier\nSaved model and config.")
+
+
+
+AutoConfig.register("scBloodClassifier", scBloodClassifierConfig)
+
+AutoModel.register(scBloodClassifierConfig, scBloodClassifier)