@huggingface/tasks 0.11.12 → 0.12.0

package/src/tasks/text-to-image/spec/output.json

@@ -6,7 +6,7 @@
 	"type": "object",
 	"properties": {
 		"image": {
-			"description": "The generated image"
+			"description": "The generated image returned as raw bytes in the payload."
 		}
 	},
 	"required": ["image"]

package/src/tasks/translation/inference.ts

@@ -6,43 +6,54 @@
 
 /**
  * Inputs for Translation inference
- *
- * Inputs for Text2text Generation inference
  */
 export interface TranslationInput {
 	/**
-	 * The input text data
+	 * The text to translate.
 	 */
 	inputs: string;
 	/**
 	 * Additional inference parameters
 	 */
-	parameters?: Text2TextGenerationParameters;
+	parameters?: TranslationParameters;
 	[property: string]: unknown;
 }
 
 /**
  * Additional inference parameters
  *
- * Additional inference parameters for Text2text Generation
+ * Additional inference parameters for Translation
  */
-export interface Text2TextGenerationParameters {
+export interface TranslationParameters {
 	/**
 	 * Whether to clean up the potential extra spaces in the text output.
 	 */
 	clean_up_tokenization_spaces?: boolean;
 	/**
-	 * Additional parametrization of the text generation algorithm
+	 * Additional parametrization of the text generation algorithm.
 	 */
 	generate_parameters?: { [key: string]: unknown };
 	/**
-	 * The truncation strategy to use
+	 * The source language of the text. Required for models that can translate from multiple
+	 * languages.
+	 */
+	src_lang?: string;
+	/**
+	 * Target language to translate to. Required for models that can translate to multiple
+	 * languages.
 	 */
-	truncation?: Text2TextGenerationTruncationStrategy;
+	tgt_lang?: string;
+	/**
+	 * The truncation strategy to use.
+	 */
+	truncation?: TranslationTruncationStrategy;
 	[property: string]: unknown;
 }
 
-export type Text2TextGenerationTruncationStrategy = "do_not_truncate" | "longest_first" | "only_first" | "only_second";
+/**
+ * The truncation strategy to use.
+ */
+export type TranslationTruncationStrategy = "do_not_truncate" | "longest_first" | "only_first" | "only_second";
 
 /**
  * Outputs of inference for the Translation task

package/src/tasks/translation/spec/input.json

@@ -1,7 +1,50 @@
 {
-	"$ref": "/inference/schemas/text2text-generation/input.json",
 	"$id": "/inference/schemas/translation/input.json",
 	"$schema": "http://json-schema.org/draft-06/schema#",
+	"description": "Inputs for Translation inference",
 	"title": "TranslationInput",
-	"description": "Inputs for Translation inference"
+	"type": "object",
+	"properties": {
+		"inputs": {
+			"description": "The text to translate.",
+			"type": "string"
+		},
+		"parameters": {
+			"description": "Additional inference parameters",
+			"$ref": "#/$defs/TranslationParameters"
+		}
+	},
+	"$defs": {
+		"TranslationParameters": {
+			"title": "TranslationParameters",
+			"description": "Additional inference parameters for Translation",
+			"type": "object",
+			"properties": {
+				"src_lang": {
+					"type": "string",
+					"description": "The source language of the text. Required for models that can translate from multiple languages."
+				},
+				"tgt_lang": {
+					"type": "string",
+					"description": "Target language to translate to. Required for models that can translate to multiple languages."
+				},
+				"clean_up_tokenization_spaces": {
+					"type": "boolean",
+					"description": "Whether to clean up the potential extra spaces in the text output."
+				},
+				"truncation": {
+					"title": "TranslationTruncationStrategy",
+					"type": "string",
+					"description": "The truncation strategy to use.",
+					"enum": ["do_not_truncate", "longest_first", "only_first", "only_second"]
+				},
+				"generate_parameters": {
+					"title": "generateParameters",
+					"type": "object",
+					"description": "Additional parametrization of the text generation algorithm."
+				}
+			}
+		}
+	},
+	"required": ["inputs"]
 }

package/src/tasks/zero-shot-classification/spec/output.json

@@ -5,6 +5,7 @@
 	"title": "ZeroShotClassificationOutput",
 	"type": "array",
 	"items": {
+		"type": "object",
 		"$ref": "/inference/schemas/common-definitions.json#/definitions/ClassificationOutput"
 	}
 }