DEBUG

app.py

@@ -136,8 +136,8 @@ def process_func(x: np.ndarray, sampling_rate: int) -> dict:
         # run through model
         with torch.no_grad():
             y = model(y)
-            print(f"{y.shape=}")
-            if y.shape[0] == 2:
+            print(f"{len(y)=}")
+            if len(y) == 2:
                 # Age-gender model
                 y = torch.hstack([y[1], y[2]])
             else: