Move model files to root

.gitattributes

@@ -33,4 +33,3 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-trained_model/model.safetensors filter=lfs diff=lfs merge=lfs -text

register_pipeline.py

@@ -4,7 +4,7 @@ from transformers.pipelines import PIPELINE_REGISTRY
 from bert_paper_classifier import SciBertPaperClassifierPipeline
 
 PIPELINE_REGISTRY.register_pipeline(
-    "text-classification",
+    "paper-classification",
     pipeline_class=SciBertPaperClassifierPipeline,
     pt_model=AutoModelForSequenceClassification,
 )

trained_model/config.json

@@ -1,103 +0,0 @@
-{
-  "architectures": [
-    "BertForSequenceClassification"
-  ],
-  "attention_probs_dropout_prob": 0.1,
-  "classifier_dropout": null,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.1,
-  "hidden_size": 768,
-  "id2label": {
-    "0": "acc-phys",
-    "1": "adap-org",
-    "2": "alg-geom",
-    "3": "ao-sci",
-    "4": "astro-ph",
-    "5": "atom-ph",
-    "6": "chao-dyn",
-    "7": "chem-ph",
-    "8": "cmp-lg",
-    "9": "comp-gas",
-    "10": "cond-mat",
-    "11": "cs",
-    "12": "dg-ga",
-    "13": "econ",
-    "14": "eess",
-    "15": "funct-an",
-    "16": "gr-qc",
-    "17": "hep-ex",
-    "18": "hep-lat",
-    "19": "hep-ph",
-    "20": "hep-th",
-    "21": "math",
-    "22": "math-ph",
-    "23": "mtrl-th",
-    "24": "nlin",
-    "25": "nucl-ex",
-    "26": "nucl-th",
-    "27": "patt-sol",
-    "28": "physics",
-    "29": "plasm-ph",
-    "30": "q-alg",
-    "31": "q-bio",
-    "32": "q-fin",
-    "33": "quant-ph",
-    "34": "solv-int",
-    "35": "stat",
-    "36": "supr-con"
-  },
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "label2id": {
-    "acc-phys": 0,
-    "adap-org": 1,
-    "alg-geom": 2,
-    "ao-sci": 3,
-    "astro-ph": 4,
-    "atom-ph": 5,
-    "chao-dyn": 6,
-    "chem-ph": 7,
-    "cmp-lg": 8,
-    "comp-gas": 9,
-    "cond-mat": 10,
-    "cs": 11,
-    "dg-ga": 12,
-    "econ": 13,
-    "eess": 14,
-    "funct-an": 15,
-    "gr-qc": 16,
-    "hep-ex": 17,
-    "hep-lat": 18,
-    "hep-ph": 19,
-    "hep-th": 20,
-    "math": 21,
-    "math-ph": 22,
-    "mtrl-th": 23,
-    "nlin": 24,
-    "nucl-ex": 25,
-    "nucl-th": 26,
-    "patt-sol": 27,
-    "physics": 28,
-    "plasm-ph": 29,
-    "q-alg": 30,
-    "q-bio": 31,
-    "q-fin": 32,
-    "quant-ph": 33,
-    "solv-int": 34,
-    "stat": 35,
-    "supr-con": 36
-  },
-  "layer_norm_eps": 1e-12,
-  "max_position_embeddings": 512,
-  "model_type": "bert",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
-  "pad_token_id": 0,
-  "position_embedding_type": "absolute",
-  "problem_type": "single_label_classification",
-  "torch_dtype": "float32",
-  "transformers_version": "4.51.0",
-  "type_vocab_size": 2,
-  "use_cache": true,
-  "vocab_size": 31090
-}