Upload folder using huggingface_hub

dnabert2_exon_intron_classification.py

@@ -7,7 +7,7 @@ from transformers.utils.generic import ModelOutput
 
 
 def process_label(p: str) -> str:
-	return "EXON" if p == 0 else "INTRON"
+	return "INTRON" if p == 0 else "EXON"
 
 class DNABERT2ExonIntronClassificationPipeline(Pipeline):
 	def _sanitize_parameters(
@@ -58,7 +58,6 @@ class DNABERT2ExonIntronClassificationPipeline(Pipeline):
 		return {"prompt": sequence, "inputs": enc}
 	
 	def _forward(self, input_tensors: dict, **forward_params):
-		assert isinstance(self.model, BertForSequenceClassification)
 		kwargs = dict(forward_params)
 
 		inputs = input_tensors.get("inputs")