Update FeatureExtractionWidget.svelte

osanseviero · web-flow · commit a8493df033bb · 2021-06-25T16:57:28.000+02:00
diff --git a/widgets/src/lib/InferenceWidget/widgets/FeatureExtractionWidget/FeatureExtractionWidget.svelte b/widgets/src/lib/InferenceWidget/widgets/FeatureExtractionWidget/FeatureExtractionWidget.svelte
@@ -145,10 +145,6 @@
 		if (body.length === 1) {
 			body = body[0];
 		}
-		if (model.modelId.startsWith(`sentence-`)) {
-			/// Dev(while correct pooling isn't applied)
-			body = body[0];
-		}
 		return new DataTable(body);
 	}
 

Original file line number	Diff line number	Diff line change
`@@ -145,10 +145,6 @@`
`145`	`145`	`if (body.length === 1) {`
`146`	`146`	`body = body[0];`
`147`	`147`	`}`
`148`		- if (model.modelId.startsWith(`sentence-`)) {
`149`		`- /// Dev(while correct pooling isn't applied)`
`150`		`- body = body[0];`
`151`		`- }`
`152`	`148`	`return new DataTable(body);`
`153`	`149`	`}`
`154`	`150`