Update app.py

app.py

@@ -37,10 +37,10 @@ def save_audio(samples: torch.Tensor):
     sample_rate = 30000
     save_path = "/tmp/audio_output"  # Use /tmp directory
 
-    # if not os.path.exists(save_path):
-    #     os.makedirs(save_path)
+    if not os.path.exists(save_path):
+        os.makedirs(save_path)
 
-    # assert samples.dim() == 2 or samples.dim() == 3
+    assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
     if samples.dim() == 2: