Upload folder using huggingface_hub

README.md

@@ -104,14 +104,10 @@ class ASJCMultiLabelPipeline(TextClassificationPipeline):
         # Convert logits to probabilities using sigmoid
         scores = torch.sigmoid(torch.tensor(model_outputs["logits"])).tolist()
 
-        # Handle single-input batch: take the first element
-        if isinstance(scores[0], list):
-            scores = scores[0]
-
         results = []
-        for i, score in enumerate(scores):
+        for i, score in enumerate(scores[0]):
             if score >= self.threshold:
-                label = self.model.config.id2label[str(i)]
+                label = self.model.config.id2label[(i)]
                 results.append({"label": label, "score": float(score)})
 
         # Sort by descending score