Update app.py

app.py

@@ -225,7 +225,10 @@ def load_spec_for_manual_masking(audio_path):
     norm_spec = normalize(norm_spec) # normalize to [-1, 1], because pipeline do not normalize for torch.Tensor input
     
     raw_image = image_add_color(torch_to_pil(norm_spec))
-    return raw_image
+
+    input_spec_image_path = "input_spectrogram.png"
+    raw_image.save(input_spec_image_path)
+    return input_spec_image_path
     
 
 css="""