Update test.py

test.py

@@ -135,7 +135,7 @@ def test(args):
             print(interp_anomaly_map.min(), interp_anomaly_map.max())
             dic = {
                 "img_level_score": text_probs.item(),
-                "pix_level_score": interp_anomaly_map.squeeze(0).cpu().detach().numpy()
+                "pix_level_score": interp_anomaly_map.cpu().detach().numpy()
             }
             os.makedirs(os.path.dirname(new_path), exist_ok=True)
             dict_to_json(dic, new_path)