Upload folder using huggingface_hub

pipeline.py

@@ -22,4 +22,11 @@ class SpanClassificationPipeline(Pipeline):
 
     def postprocess(self, model_outputs):
         logits = model_outputs.logits
-        return int(torch.argmax(logits, dim=1).item())
+        return int(torch.argmax(logits, dim=1).item())
+
+
+PIPELINE_REGISTRY.register_pipeline(
+    task="spancnn-classification",
+    pipeline_class=SpanClassificationPipeline,  
+    pt_model=AutoModelForSequenceClassification, 
+)