Fix zero-shot-object-detection percentage option (#434)

huggingface · Dec 6, 2023 · d318b1f · d318b1f
1 parent c0c746b
commit d318b1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pipelines.js b/src/pipelines.js
@@ -1989,7 +1989,7 @@ export class ZeroShotObjectDetectionPipeline extends Pipeline {
         const toReturn = [];
         for (let i = 0; i < images.length; ++i) {
             const image = images[i];
-            const imageSize = [[image.height, image.width]];
+            const imageSize = percentage ? null : [[image.height, image.width]];
             const pixel_values = model_inputs.pixel_values[i].unsqueeze_(0);
 
             // Run model with both text and pixel inputs