From 72b6db19d695b6272b2655c39843e7d61d565384 Mon Sep 17 00:00:00 2001
From: h-alice <admin@halice.art>
Date: Mon, 13 Jan 2025 23:35:48 +0800
Subject: [PATCH] patch: fix dimension

---
 deepface/models/Demography.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/deepface/models/Demography.py b/deepface/models/Demography.py
index 2a78afd..c9cfe30 100644
--- a/deepface/models/Demography.py
+++ b/deepface/models/Demography.py
@@ -40,7 +40,6 @@ class Demography(ABC):
         assert img_batch.ndim == 4, "expected 4-dimensional tensor input"
                   
         if img_batch.shape[0] == 1: # Single image
-            img_batch = img_batch.squeeze(0) # Remove batch dimension
             # Predict with legacy method.
             return self.model(img_batch, training=False).numpy()[0, :]
         else: