@huggingface/inference 3.0.1 → 3.1.0

package/dist/index.js

@@ -61,6 +61,10 @@ var FAL_AI_SUPPORTED_MODEL_IDS = {
   },
   "automatic-speech-recognition": {
     "openai/whisper-large-v3": "fal-ai/whisper"
+  },
+  "text-to-video": {
+    "genmo/mochi-1-preview": "fal-ai/mochi-v1",
+    "tencent/HunyuanVideo": "fal-ai/hunyuan-video"
   }
 };
 
@@ -73,6 +77,9 @@ var REPLICATE_SUPPORTED_MODEL_IDS = {
   },
   "text-to-speech": {
     "OuteAI/OuteTTS-0.3-500M": "jbilcke/oute-tts:39a59319327b27327fa3095149c5a746e7f2aee18c75055c3368237a6503cd26"
+  },
+  "text-to-video": {
+    "genmo/mochi-1-preview": "genmoai/mochi-1:1944af04d098ef69bed7f9d335d102e652203f268ec4aaa2d836f6217217e460"
   }
 };
 
@@ -538,9 +545,42 @@ var InferenceOutputError = class extends TypeError {
   }
 };
 
+// src/utils/pick.ts
+function pick(o, props) {
+  return Object.assign(
+    {},
+    ...props.map((prop) => {
+      if (o[prop] !== void 0) {
+        return { [prop]: o[prop] };
+      }
+    })
+  );
+}
+
+// src/utils/typedInclude.ts
+function typedInclude(arr, v) {
+  return arr.includes(v);
+}
+
+// src/utils/omit.ts
+function omit(o, props) {
+  const propsArr = Array.isArray(props) ? props : [props];
+  const letsKeep = Object.keys(o).filter((prop) => !typedInclude(propsArr, prop));
+  return pick(o, letsKeep);
+}
+
+// src/tasks/audio/utils.ts
+function preparePayload(args) {
+  return "data" in args ? args : {
+    ...omit(args, "inputs"),
+    data: args.inputs
+  };
+}
+
 // src/tasks/audio/audioClassification.ts
 async function audioClassification(args, options) {
-  const res = await request(args, {
+  const payload = preparePayload(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "audio-classification"
   });
@@ -566,15 +606,8 @@ function base64FromBytes(arr) {
 
 // src/tasks/audio/automaticSpeechRecognition.ts
 async function automaticSpeechRecognition(args, options) {
-  if (args.provider === "fal-ai") {
-    const contentType = args.data instanceof Blob ? args.data.type : "audio/mpeg";
-    const base64audio = base64FromBytes(
-      new Uint8Array(args.data instanceof ArrayBuffer ? args.data : await args.data.arrayBuffer())
-    );
-    args.audio_url = `data:${contentType};base64,${base64audio}`;
-    delete args.data;
-  }
-  const res = await request(args, {
+  const payload = await buildPayload(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "automatic-speech-recognition"
   });
@@ -584,6 +617,32 @@ async function automaticSpeechRecognition(args, options) {
   }
   return res;
 }
+var FAL_AI_SUPPORTED_BLOB_TYPES = ["audio/mpeg", "audio/mp4", "audio/wav", "audio/x-wav"];
+async function buildPayload(args) {
+  if (args.provider === "fal-ai") {
+    const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : void 0;
+    const contentType = blob?.type;
+    if (!contentType) {
+      throw new Error(
+        `Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
+      );
+    }
+    if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
+      throw new Error(
+        `Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
+          ", "
+        )}`
+      );
+    }
+    const base64audio = base64FromBytes(new Uint8Array(await blob.arrayBuffer()));
+    return {
+      ..."data" in args ? omit(args, "data") : omit(args, "inputs"),
+      audio_url: `data:${contentType};base64,${base64audio}`
+    };
+  } else {
+    return preparePayload(args);
+  }
+}
 
 // src/tasks/audio/textToSpeech.ts
 async function textToSpeech(args, options) {
@@ -591,6 +650,9 @@ async function textToSpeech(args, options) {
     ...options,
     taskHint: "text-to-speech"
   });
+  if (res instanceof Blob) {
+    return res;
+  }
   if (res && typeof res === "object") {
     if ("output" in res) {
       if (typeof res.output === "string") {
@@ -604,31 +666,39 @@ async function textToSpeech(args, options) {
       }
     }
   }
-  const isValidOutput = res && res instanceof Blob;
-  if (!isValidOutput) {
-    throw new InferenceOutputError("Expected Blob");
-  }
-  return res;
+  throw new InferenceOutputError("Expected Blob or object with output");
 }
 
 // src/tasks/audio/audioToAudio.ts
 async function audioToAudio(args, options) {
-  const res = await request(args, {
+  const payload = preparePayload(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "audio-to-audio"
   });
-  const isValidOutput = Array.isArray(res) && res.every(
-    (x) => typeof x.label === "string" && typeof x.blob === "string" && typeof x["content-type"] === "string"
-  );
-  if (!isValidOutput) {
-    throw new InferenceOutputError("Expected Array<{label: string, blob: string, content-type: string}>");
+  return validateOutput(res);
+}
+function validateOutput(output) {
+  if (!Array.isArray(output)) {
+    throw new InferenceOutputError("Expected Array");
   }
-  return res;
+  if (!output.every((elem) => {
+    return typeof elem === "object" && elem && "label" in elem && typeof elem.label === "string" && "content-type" in elem && typeof elem["content-type"] === "string" && "blob" in elem && typeof elem.blob === "string";
+  })) {
+    throw new InferenceOutputError("Expected Array<{label: string, audio: Blob}>");
+  }
+  return output;
+}
+
+// src/tasks/cv/utils.ts
+function preparePayload2(args) {
+  return "data" in args ? args : { ...omit(args, "inputs"), data: args.inputs };
 }
 
 // src/tasks/cv/imageClassification.ts
 async function imageClassification(args, options) {
-  const res = await request(args, {
+  const payload = preparePayload2(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "image-classification"
   });
@@ -641,7 +711,8 @@ async function imageClassification(args, options) {
 
 // src/tasks/cv/imageSegmentation.ts
 async function imageSegmentation(args, options) {
-  const res = await request(args, {
+  const payload = preparePayload2(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "image-segmentation"
   });
@@ -654,7 +725,8 @@ async function imageSegmentation(args, options) {
 
 // src/tasks/cv/imageToText.ts
 async function imageToText(args, options) {
-  const res = (await request(args, {
+  const payload = preparePayload2(args);
+  const res = (await request(payload, {
     ...options,
     taskHint: "image-to-text"
   }))?.[0];
@@ -666,7 +738,8 @@ async function imageToText(args, options) {
 
 // src/tasks/cv/objectDetection.ts
 async function objectDetection(args, options) {
-  const res = await request(args, {
+  const payload = preparePayload2(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "object-detection"
   });
@@ -683,15 +756,13 @@ async function objectDetection(args, options) {
 
 // src/tasks/cv/textToImage.ts
 async function textToImage(args, options) {
-  if (args.provider === "together" || args.provider === "fal-ai") {
-    args.prompt = args.inputs;
-    delete args.inputs;
-    args.response_format = "base64";
-  } else if (args.provider === "replicate") {
-    args.prompt = args.inputs;
-    delete args.inputs;
-  }
-  const res = await request(args, {
+  const payload = args.provider === "together" || args.provider === "fal-ai" || args.provider === "replicate" ? {
+    ...omit(args, ["inputs", "parameters"]),
+    ...args.parameters,
+    ...args.provider !== "replicate" ? { response_format: "base64" } : void 0,
+    prompt: args.inputs
+  } : args;
+  const res = await request(payload, {
     ...options,
     taskHint: "text-to-image"
   });
@@ -748,18 +819,30 @@ async function imageToImage(args, options) {
 }
 
 // src/tasks/cv/zeroShotImageClassification.ts
-async function zeroShotImageClassification(args, options) {
-  const reqArgs = {
-    ...args,
-    inputs: {
-      image: base64FromBytes(
-        new Uint8Array(
-          args.inputs.image instanceof ArrayBuffer ? args.inputs.image : await args.inputs.image.arrayBuffer()
+async function preparePayload3(args) {
+  if (args.inputs instanceof Blob) {
+    return {
+      ...args,
+      inputs: {
+        image: base64FromBytes(new Uint8Array(await args.inputs.arrayBuffer()))
+      }
+    };
+  } else {
+    return {
+      ...args,
+      inputs: {
+        image: base64FromBytes(
+          new Uint8Array(
+            args.inputs.image instanceof ArrayBuffer ? args.inputs.image : await args.inputs.image.arrayBuffer()
+          )
         )
-      )
-    }
-  };
-  const res = await request(reqArgs, {
+      }
+    };
+  }
+}
+async function zeroShotImageClassification(args, options) {
+  const payload = await preparePayload3(args);
+  const res = await request(payload, {
     ...options,
     taskHint: "zero-shot-image-classification"
   });
@@ -848,17 +931,19 @@ async function questionAnswering(args, options) {
     ...options,
     taskHint: "question-answering"
   });
-  const isValidOutput = typeof res === "object" && !!res && typeof res.answer === "string" && typeof res.end === "number" && typeof res.score === "number" && typeof res.start === "number";
+  const isValidOutput = Array.isArray(res) ? res.every(
+    (elem) => typeof elem === "object" && !!elem && typeof elem.answer === "string" && typeof elem.end === "number" && typeof elem.score === "number" && typeof elem.start === "number"
+  ) : typeof res === "object" && !!res && typeof res.answer === "string" && typeof res.end === "number" && typeof res.score === "number" && typeof res.start === "number";
   if (!isValidOutput) {
-    throw new InferenceOutputError("Expected {answer: string, end: number, score: number, start: number}");
+    throw new InferenceOutputError("Expected Array<{answer: string, end: number, score: number, start: number}>");
   }
-  return res;
+  return Array.isArray(res) ? res[0] : res;
 }
 
 // src/tasks/nlp/sentenceSimilarity.ts
 async function sentenceSimilarity(args, options) {
   const defaultTask = args.model ? await getDefaultTask(args.model, args.accessToken, options) : void 0;
-  const res = await request(args, {
+  const res = await request(prepareInput(args), {
     ...options,
     taskHint: "sentence-similarity",
     ...defaultTask === "feature-extraction" && { forceTask: "sentence-similarity" }
@@ -869,6 +954,13 @@ async function sentenceSimilarity(args, options) {
   }
   return res;
 }
+function prepareInput(args) {
+  return {
+    ...omit(args, ["inputs", "parameters"]),
+    inputs: { ...omit(args.inputs, "sourceSentence") },
+    parameters: { source_sentence: args.inputs.sourceSentence, ...args.parameters }
+  };
+}
 
 // src/tasks/nlp/summarization.ts
 async function summarization(args, options) {
@@ -889,13 +981,18 @@ async function tableQuestionAnswering(args, options) {
     ...options,
     taskHint: "table-question-answering"
   });
-  const isValidOutput = typeof res?.aggregator === "string" && typeof res.answer === "string" && Array.isArray(res.cells) && res.cells.every((x) => typeof x === "string") && Array.isArray(res.coordinates) && res.coordinates.every((coord) => Array.isArray(coord) && coord.every((x) => typeof x === "number"));
+  const isValidOutput = Array.isArray(res) ? res.every((elem) => validate(elem)) : validate(res);
   if (!isValidOutput) {
     throw new InferenceOutputError(
       "Expected {aggregator: string, answer: string, cells: string[], coordinates: number[][]}"
     );
   }
-  return res;
+  return Array.isArray(res) ? res[0] : res;
+}
+function validate(elem) {
+  return typeof elem === "object" && !!elem && "aggregator" in elem && typeof elem.aggregator === "string" && "answer" in elem && typeof elem.answer === "string" && "cells" in elem && Array.isArray(elem.cells) && elem.cells.every((x) => typeof x === "string") && "coordinates" in elem && Array.isArray(elem.coordinates) && elem.coordinates.every(
+    (coord) => Array.isArray(coord) && coord.every((x) => typeof x === "number")
+  );
 }
 
 // src/tasks/nlp/textClassification.ts
@@ -1038,11 +1135,7 @@ async function documentQuestionAnswering(args, options) {
     inputs: {
       question: args.inputs.question,
       // convert Blob or ArrayBuffer to base64
-      image: base64FromBytes(
-        new Uint8Array(
-          args.inputs.image instanceof ArrayBuffer ? args.inputs.image : await args.inputs.image.arrayBuffer()
-        )
-      )
+      image: base64FromBytes(new Uint8Array(await args.inputs.image.arrayBuffer()))
     }
   };
   const res = toArray(
@@ -1050,12 +1143,14 @@ async function documentQuestionAnswering(args, options) {
       ...options,
       taskHint: "document-question-answering"
     })
-  )?.[0];
-  const isValidOutput = typeof res?.answer === "string" && (typeof res.end === "number" || typeof res.end === "undefined") && (typeof res.score === "number" || typeof res.score === "undefined") && (typeof res.start === "number" || typeof res.start === "undefined");
+  );
+  const isValidOutput = Array.isArray(res) && res.every(
+    (elem) => typeof elem === "object" && !!elem && typeof elem?.answer === "string" && (typeof elem.end === "number" || typeof elem.end === "undefined") && (typeof elem.score === "number" || typeof elem.score === "undefined") && (typeof elem.start === "number" || typeof elem.start === "undefined")
+  );
   if (!isValidOutput) {
     throw new InferenceOutputError("Expected Array<{answer: string, end?: number, score?: number, start?: number}>");
   }
-  return res;
+  return res[0];
 }
 
 // src/tasks/multimodal/visualQuestionAnswering.ts
@@ -1065,22 +1160,20 @@ async function visualQuestionAnswering(args, options) {
     inputs: {
       question: args.inputs.question,
       // convert Blob or ArrayBuffer to base64
-      image: base64FromBytes(
-        new Uint8Array(
-          args.inputs.image instanceof ArrayBuffer ? args.inputs.image : await args.inputs.image.arrayBuffer()
-        )
-      )
+      image: base64FromBytes(new Uint8Array(await args.inputs.image.arrayBuffer()))
     }
   };
-  const res = (await request(reqArgs, {
+  const res = await request(reqArgs, {
     ...options,
     taskHint: "visual-question-answering"
-  }))?.[0];
-  const isValidOutput = typeof res?.answer === "string" && typeof res.score === "number";
+  });
+  const isValidOutput = Array.isArray(res) && res.every(
+    (elem) => typeof elem === "object" && !!elem && typeof elem?.answer === "string" && typeof elem.score === "number"
+  );
   if (!isValidOutput) {
     throw new InferenceOutputError("Expected Array<{answer: string, score: number}>");
   }
-  return res;
+  return res[0];
 }
 
 // src/tasks/tabular/tabularRegression.ts

package/dist/src/providers/fal-ai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fal-ai.d.ts","sourceRoot":"","sources":["../../../src/providers/fal-ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE/C,eAAO,MAAM,mBAAmB,oBAAoB,CAAC;AAErD,KAAK,OAAO,GAAG,MAAM,CAAC;AAEtB,eAAO,MAAM,0BAA0B,EAAE,eAAe,CAAC,OAAO,CAgB/D,CAAC"}
+{"version":3,"file":"fal-ai.d.ts","sourceRoot":"","sources":["../../../src/providers/fal-ai.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE/C,eAAO,MAAM,mBAAmB,oBAAoB,CAAC;AAErD,KAAK,OAAO,GAAG,MAAM,CAAC;AAEtB,eAAO,MAAM,0BAA0B,EAAE,eAAe,CAAC,OAAO,CAoB/D,CAAC"}

package/dist/src/providers/replicate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"replicate.d.ts","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE/C,eAAO,MAAM,sBAAsB,8BAA8B,CAAC;AAElE,KAAK,WAAW,GAAG,MAAM,CAAC;AAE1B,eAAO,MAAM,6BAA6B,EAAE,eAAe,CAAC,WAAW,CAStE,CAAC"}
+{"version":3,"file":"replicate.d.ts","sourceRoot":"","sources":["../../../src/providers/replicate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE/C,eAAO,MAAM,sBAAsB,8BAA8B,CAAC;AAElE,KAAK,WAAW,GAAG,MAAM,CAAC;AAE1B,eAAO,MAAM,6BAA6B,EAAE,eAAe,CAAC,WAAW,CAYtE,CAAC"}

package/dist/src/tasks/audio/audioClassification.d.ts

@@ -1,24 +1,10 @@
+import type { AudioClassificationInput, AudioClassificationOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type AudioClassificationArgs = BaseArgs & {
-    /**
-     * Binary audio data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface AudioClassificationOutputValue {
-    /**
-     * The label for the class (model specific)
-     */
-    label: string;
-    /**
-     * A float that represents how likely it is that the audio file belongs to this class.
-     */
-    score: number;
-}
-export type AudioClassificationReturn = AudioClassificationOutputValue[];
+import type { LegacyAudioInput } from "./utils";
+export type AudioClassificationArgs = BaseArgs & (AudioClassificationInput | LegacyAudioInput);
 /**
  * This task reads some audio input and outputs the likelihood of classes.
  * Recommended model:  superb/hubert-large-superb-er
  */
-export declare function audioClassification(args: AudioClassificationArgs, options?: Options): Promise<AudioClassificationReturn>;
+export declare function audioClassification(args: AudioClassificationArgs, options?: Options): Promise<AudioClassificationOutput>;
 //# sourceMappingURL=audioClassification.d.ts.map

package/dist/src/tasks/audio/audioClassification.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"audioClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/audioClassification.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG;IAChD;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,8BAA8B;IAC9C;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,yBAAyB,GAAG,8BAA8B,EAAE,CAAC;AAEzE;;;GAGG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CAWpC"}
+{"version":3,"file":"audioClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/audioClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE9F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAGhD,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG,CAAC,wBAAwB,GAAG,gBAAgB,CAAC,CAAC;AAE/F;;;GAGG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CAYpC"}

package/dist/src/tasks/audio/audioToAudio.d.ts

@@ -1,11 +1,12 @@
 import type { BaseArgs, Options } from "../../types";
-export type AudioToAudioArgs = BaseArgs & {
+import type { LegacyAudioInput } from "./utils";
+export type AudioToAudioArgs = (BaseArgs & {
     /**
      * Binary audio data
      */
-    data: Blob | ArrayBuffer;
-};
-export interface AudioToAudioOutputValue {
+    inputs: Blob;
+}) | LegacyAudioInput;
+export interface AudioToAudioOutputElem {
     /**
      * The label for the audio output (model specific)
      */
@@ -13,16 +14,16 @@ export interface AudioToAudioOutputValue {
     /**
      * Base64 encoded audio output.
      */
+    audio: Blob;
+}
+export interface AudioToAudioOutput {
     blob: string;
-    /**
-     * Content-type for blob, e.g. audio/flac
-     */
     "content-type": string;
+    label: string;
 }
-export type AudioToAudioReturn = AudioToAudioOutputValue[];
 /**
  * This task reads some audio input and outputs one or multiple audio files.
  * Example model: speechbrain/sepformer-wham does audio source separation.
  */
-export declare function audioToAudio(args: AudioToAudioArgs, options?: Options): Promise<AudioToAudioReturn>;
+export declare function audioToAudio(args: AudioToAudioArgs, options?: Options): Promise<AudioToAudioOutput[]>;
 //# sourceMappingURL=audioToAudio.d.ts.map

package/dist/src/tasks/audio/audioToAudio.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"audioToAudio.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/audioToAudio.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG;IACzC;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,uBAAuB;IACvC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IAEb;;OAEG;IACH,cAAc,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,MAAM,kBAAkB,GAAG,uBAAuB,EAAE,CAAC;AAE3D;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAczG"}
+{"version":3,"file":"audioToAudio.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/audioToAudio.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAGhD,MAAM,MAAM,gBAAgB,GACzB,CAAC,QAAQ,GAAG;IACZ;;OAEG;IACH,MAAM,EAAE,IAAI,CAAC;CACZ,CAAC,GACF,gBAAgB,CAAC;AAEpB,MAAM,WAAW,sBAAsB;IACtC;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IAEd;;OAEG;IACH,KAAK,EAAE,IAAI,CAAC;CACZ;AAED,MAAM,WAAW,kBAAkB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;CACd;AAED;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC,CAQ3G"}

package/dist/src/tasks/audio/automaticSpeechRecognition.d.ts

@@ -1,16 +1,7 @@
+import type { AutomaticSpeechRecognitionInput, AutomaticSpeechRecognitionOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type AutomaticSpeechRecognitionArgs = BaseArgs & {
-    /**
-     * Binary audio data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface AutomaticSpeechRecognitionOutput {
-    /**
-     * The text that was recognized from the audio
-     */
-    text: string;
-}
+import type { LegacyAudioInput } from "./utils";
+export type AutomaticSpeechRecognitionArgs = BaseArgs & (AutomaticSpeechRecognitionInput | LegacyAudioInput);
 /**
  * This task reads some audio input and outputs the said words within the audio files.
  * Recommended model (english language): facebook/wav2vec2-large-960h-lv60-self

package/dist/src/tasks/audio/automaticSpeechRecognition.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"automaticSpeechRecognition.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/automaticSpeechRecognition.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAIlE,MAAM,MAAM,8BAA8B,GAAG,QAAQ,GAAG;IACvD;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,gCAAgC;IAChD;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;CACb;AAED;;;GAGG;AACH,wBAAsB,0BAA0B,CAC/C,IAAI,EAAE,8BAA8B,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,gCAAgC,CAAC,CAkB3C"}
+{"version":3,"file":"automaticSpeechRecognition.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/automaticSpeechRecognition.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,+BAA+B,EAAE,gCAAgC,EAAE,MAAM,oBAAoB,CAAC;AAE5G,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAe,MAAM,aAAa,CAAC;AAGlE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAIhD,MAAM,MAAM,8BAA8B,GAAG,QAAQ,GAAG,CAAC,+BAA+B,GAAG,gBAAgB,CAAC,CAAC;AAC7G;;;GAGG;AACH,wBAAsB,0BAA0B,CAC/C,IAAI,EAAE,8BAA8B,EACpC,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,gCAAgC,CAAC,CAW3C"}

package/dist/src/tasks/audio/textToSpeech.d.ts

@@ -1,14 +1,10 @@
+import type { TextToSpeechInput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type TextToSpeechArgs = BaseArgs & {
-    /**
-     * The text to generate an audio from
-     */
-    inputs: string;
-};
-export type TextToSpeechOutput = Blob;
+type TextToSpeechArgs = BaseArgs & TextToSpeechInput;
 /**
  * This task synthesize an audio of a voice pronouncing a given text.
  * Recommended model: espnet/kan-bayashi_ljspeech_vits
  */
-export declare function textToSpeech(args: TextToSpeechArgs, options?: Options): Promise<TextToSpeechOutput>;
+export declare function textToSpeech(args: TextToSpeechArgs, options?: Options): Promise<Blob>;
+export {};
 //# sourceMappingURL=textToSpeech.d.ts.map

package/dist/src/tasks/audio/textToSpeech.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"textToSpeech.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/textToSpeech.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,gBAAgB,GAAG,QAAQ,GAAG;IACzC;;OAEG;IACH,MAAM,EAAE,MAAM,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,IAAI,CAAC;AAItC;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAuBzG"}
+{"version":3,"file":"textToSpeech.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/textToSpeech.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAE5D,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,KAAK,gBAAgB,GAAG,QAAQ,GAAG,iBAAiB,CAAC;AAKrD;;;GAGG;AACH,wBAAsB,YAAY,CAAC,IAAI,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAsB3F"}

package/dist/src/tasks/audio/utils.d.ts

@@ -0,0 +1,11 @@
+import type { BaseArgs, RequestArgs } from "../../types";
+/**
+ * @deprecated
+ */
+export interface LegacyAudioInput {
+    data: Blob | ArrayBuffer;
+}
+export declare function preparePayload(args: BaseArgs & ({
+    inputs: Blob;
+} | LegacyAudioInput)): RequestArgs;
+//# sourceMappingURL=utils.d.ts.map

package/dist/src/tasks/audio/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../../src/tasks/audio/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAGzD;;GAEG;AACH,MAAM,WAAW,gBAAgB;IAChC,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,QAAQ,GAAG,CAAC;IAAE,MAAM,EAAE,IAAI,CAAA;CAAE,GAAG,gBAAgB,CAAC,GAAG,WAAW,CAOlG"}

package/dist/src/tasks/cv/imageClassification.d.ts

@@ -1,21 +1,7 @@
+import type { ImageClassificationInput, ImageClassificationOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type ImageClassificationArgs = BaseArgs & {
-    /**
-     * Binary image data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface ImageClassificationOutputValue {
-    /**
-     * The label for the class (model specific)
-     */
-    label: string;
-    /**
-     * A float that represents how likely it is that the image file belongs to this class.
-     */
-    score: number;
-}
-export type ImageClassificationOutput = ImageClassificationOutputValue[];
+import { type LegacyImageInput } from "./utils";
+export type ImageClassificationArgs = BaseArgs & (ImageClassificationInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes.
  * Recommended model: google/vit-base-patch16-224

package/dist/src/tasks/cv/imageClassification.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"imageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageClassification.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG;IAChD;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,8BAA8B;IAC9C;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,yBAAyB,GAAG,8BAA8B,EAAE,CAAC;AAEzE;;;GAGG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CAWpC"}
+{"version":3,"file":"imageClassification.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageClassification.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,yBAAyB,EAAE,MAAM,oBAAoB,CAAC;AAE9F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEhE,MAAM,MAAM,uBAAuB,GAAG,QAAQ,GAAG,CAAC,wBAAwB,GAAG,gBAAgB,CAAC,CAAC;AAE/F;;;GAGG;AACH,wBAAsB,mBAAmB,CACxC,IAAI,EAAE,uBAAuB,EAC7B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,yBAAyB,CAAC,CAYpC"}

package/dist/src/tasks/cv/imageSegmentation.d.ts

@@ -1,25 +1,7 @@
+import type { ImageSegmentationInput, ImageSegmentationOutput } from "@huggingface/tasks";
 import type { BaseArgs, Options } from "../../types";
-export type ImageSegmentationArgs = BaseArgs & {
-    /**
-     * Binary image data
-     */
-    data: Blob | ArrayBuffer;
-};
-export interface ImageSegmentationOutputValue {
-    /**
-     * The label for the class (model specific) of a segment.
-     */
-    label: string;
-    /**
-     * A str (base64 str of a single channel black-and-white img) representing the mask of a segment.
-     */
-    mask: string;
-    /**
-     * A float that represents how likely it is that the detected object belongs to the given class.
-     */
-    score: number;
-}
-export type ImageSegmentationOutput = ImageSegmentationOutputValue[];
+import { type LegacyImageInput } from "./utils";
+export type ImageSegmentationArgs = BaseArgs & (ImageSegmentationInput | LegacyImageInput);
 /**
  * This task reads some image input and outputs the likelihood of classes & bounding boxes of detected objects.
  * Recommended model: facebook/detr-resnet-50-panoptic

package/dist/src/tasks/cv/imageSegmentation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"imageSegmentation.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageSegmentation.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAGrD,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG;IAC9C;;OAEG;IACH,IAAI,EAAE,IAAI,GAAG,WAAW,CAAC;CACzB,CAAC;AAEF,MAAM,WAAW,4BAA4B;IAC5C;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;IACd;;OAEG;IACH,IAAI,EAAE,MAAM,CAAC;IACb;;OAEG;IACH,KAAK,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,uBAAuB,GAAG,4BAA4B,EAAE,CAAC;AAErE;;;GAGG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CAYlC"}
+{"version":3,"file":"imageSegmentation.d.ts","sourceRoot":"","sources":["../../../../src/tasks/cv/imageSegmentation.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAE1F,OAAO,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAkB,KAAK,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAEhE,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,CAAC,sBAAsB,GAAG,gBAAgB,CAAC,CAAC;AAE3F;;;GAGG;AACH,wBAAsB,iBAAiB,CACtC,IAAI,EAAE,qBAAqB,EAC3B,OAAO,CAAC,EAAE,OAAO,GACf,OAAO,CAAC,uBAAuB,CAAC,CAalC"}