complete specs for translation and summarization

huggingface · Aug 28, 2024 · 701ff69 · 701ff69
1 parent f085780
commit 701ff69
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 2 deletions.
diff --git a/packages/tasks/src/tasks/summarization/spec/input.json b/packages/tasks/src/tasks/summarization/spec/input.json
@@ -3,5 +3,12 @@
 	"$id": "/inference/schemas/summarization/input.json",
 	"$schema": "http://json-schema.org/draft-06/schema#",
 	"title": "SummarizationInput",
-	"description": "Inputs for Summarization inference"
+	"description": "Inputs for Summarization inference",
+	"properties": {
+		"inputs": {
+			"description": "The text to summarize.",
+			"type": "string"
+		}
+	},
+	"required": ["inputs"]
 }
diff --git a/packages/tasks/src/tasks/translation/spec/input.json b/packages/tasks/src/tasks/translation/spec/input.json
@@ -3,5 +3,33 @@
 	"$id": "/inference/schemas/translation/input.json",
 	"$schema": "http://json-schema.org/draft-06/schema#",
 	"title": "TranslationInput",
-	"description": "Inputs for Translation inference"
+	"description": "Inputs for Translation inference",
+	"properties": {
+		"inputs": {
+			"description": "The text to translate.",
+			"type": "string"
+		},
+		"parameters": {
+			"description": "Additional inference parameters",
+			"$ref": "#/$defs/TranslationParameters"
+		}
+	},
+	"$defs": {
+		"FillMaskParameters": {
+			"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."
+				}
+			}
+		}
+	},
+	"required": ["inputs"]
 }