Bug 1918220 - replace T output a=pascalc

Original Revision: https://phabricator.services.mozilla.com/D221920 Differential Revision: https://phabricator.services.mozilla.com/D222081
i3roly · Sep 13, 2024 · 03b71d3 · 03b71d3
1 parent a7a2007
commit 03b71d3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/toolkit/components/ml/content/ONNXPipeline.mjs b/toolkit/components/ml/content/ONNXPipeline.mjs
@@ -138,6 +138,12 @@ async function imageToText(request, model, tokenizer, processor, _config) {
   }
   lazy.console.debug("Inference done in ", Date.now() - start);
   result.output = toReturn[0][0].generated_text;
+
+  // Bug 1918220 - replace the result for models with that bug
+  if (result.output === "T") {
+    lazy.console.debug("Replacing `T` with `Text document.`");
+    result.output = "Text document.";
+  }
   return result;
 }