@huggingface/inference 2.5.0 → 2.5.2

package/dist/index.d.ts

@@ -167,11 +167,11 @@ export type ImageClassificationArgs = BaseArgs & {
 export type ImageClassificationOutput = ImageClassificationOutputValue[];
 export interface ImageClassificationOutputValue {
 	/**
-	 * A float that represents how likely it is that the image file belongs to this class.
+	 * The label for the class (model specific)
 	 */
 	label: string;
 	/**
-	 * The label for the class (model specific)
+	 * A float that represents how likely it is that the image file belongs to this class.
 	 */
 	score: number;
 }
@@ -509,7 +509,7 @@ export type FeatureExtractionArgs = BaseArgs & {
 	inputs: string | string[];
 };
 /**
- * Returned values are a list of floats, or a list of list of floats, or a list of list of list of floats (depending on if you sent a string or a list of string, and if the automatic reduction, usually mean_pooling for instance was applied for you or not. This should be explained on the model's README.
+ * Returned values are a multidimensional array of floats (dimension depending on if you sent a string or a list of string, and if the automatic reduction, usually mean_pooling for instance was applied for you or not. This should be explained on the model's README).
  */
 export type FeatureExtractionOutput = (number | number[] | number[][])[];
 /**

package/dist/index.js

@@ -558,9 +558,9 @@ async function featureExtraction(args, options) {
       return arr.every((x) => typeof x === "number");
     }
   };
-  isValidOutput = Array.isArray(res) && isNumArrayRec(res, 2, 0);
+  isValidOutput = Array.isArray(res) && isNumArrayRec(res, 3, 0);
   if (!isValidOutput) {
-    throw new InferenceOutputError("Expected Array<number[][] | number[] | number>");
+    throw new InferenceOutputError("Expected Array<number[][][] | number[][] | number[] | number>");
   }
   return res;
 }

package/dist/index.mjs

@@ -506,9 +506,9 @@ async function featureExtraction(args, options) {
       return arr.every((x) => typeof x === "number");
     }
   };
-  isValidOutput = Array.isArray(res) && isNumArrayRec(res, 2, 0);
+  isValidOutput = Array.isArray(res) && isNumArrayRec(res, 3, 0);
   if (!isValidOutput) {
-    throw new InferenceOutputError("Expected Array<number[][] | number[] | number>");
+    throw new InferenceOutputError("Expected Array<number[][][] | number[][] | number[] | number>");
   }
   return res;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@huggingface/inference",
-  "version": "2.5.0",
+  "version": "2.5.2",
   "packageManager": "pnpm@8.3.1",
   "license": "MIT",
   "author": "Tim Mikeladze <tim.mikeladze@gmail.com>",

package/src/tasks/cv/imageClassification.ts

@@ -11,11 +11,11 @@ export type ImageClassificationArgs = BaseArgs & {
 
 export interface ImageClassificationOutputValue {
 	/**
-	 * A float that represents how likely it is that the image file belongs to this class.
+	 * The label for the class (model specific)
 	 */
 	label: string;
 	/**
-	 * The label for the class (model specific)
+	 * A float that represents how likely it is that the image file belongs to this class.
 	 */
 	score: number;
 }

package/src/tasks/nlp/featureExtraction.ts

@@ -14,7 +14,7 @@ export type FeatureExtractionArgs = BaseArgs & {
 };
 
 /**
- * Returned values are a list of floats, or a list of list of floats, or a list of list of list of floats (depending on if you sent a string or a list of string, and if the automatic reduction, usually mean_pooling for instance was applied for you or not. This should be explained on the model's README.
+ * Returned values are a multidimensional array of floats (dimension depending on if you sent a string or a list of string, and if the automatic reduction, usually mean_pooling for instance was applied for you or not. This should be explained on the model's README).
  */
 export type FeatureExtractionOutput = (number | number[] | number[][])[];
 
@@ -46,10 +46,10 @@ export async function featureExtraction(
 		}
 	};
 
-	isValidOutput = Array.isArray(res) && isNumArrayRec(res, 2, 0);
+	isValidOutput = Array.isArray(res) && isNumArrayRec(res, 3, 0);
 
 	if (!isValidOutput) {
-		throw new InferenceOutputError("Expected Array<number[][] | number[] | number>");
+		throw new InferenceOutputError("Expected Array<number[][][] | number[][] | number[] | number>");
 	}
 	return res;
 }