cnmoro
/

gliclass-large-v3.0-onnx

Text Classification

Model card Files Files and versions Community

cnmoro commited on 7 days ago

Commit

96286ca

·

verified ·

1 Parent(s): ed9b404

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -46,7 +46,7 @@ class GLiClassOnnxInference:
         return np.array([ids], dtype=np.int64), np.array([mask], dtype=np.int64)
     def onnx_predict(self, text: str, labels: list[str]):
-        full_text = text + "".join([f"<<LABEL>>{l}" for l in labels]) + "<<SEP>>"
         ids, mask = self.encode(full_text, max_length=512)
         ort_inputs = {"input_ids": ids, "attention_mask": mask}

         return np.array([ids], dtype=np.int64), np.array([mask], dtype=np.int64)
     def onnx_predict(self, text: str, labels: list[str]):
+        full_text = "".join([f"<<LABEL>>{l}" for l in labels]) + "<<SEP>>" + text
         ids, mask = self.encode(full_text, max_length=512)
         ort_inputs = {"input_ids": ids, "attention_mask": mask}