now supports other sample rates

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import argparse
+import copy
 
 import numpy as np
 import matplotlib.pyplot as plt
@@ -147,13 +148,19 @@ def main(input, mix_coefficient):
     meter = pyln.Meter(44100)
 
     sr, track_audio = input
+    orig_sr = copy.deepcopy(sr)
     track_audio = track_audio.T
     track_name = "gradio_demo"
 
+    if sr != 44100:
+        track_audio = int16_to_float32(track_audio)
+        track_audio = librosa.resample(
+            track_audio, orig_sr=sr, target_sr=44100, res_type="soxr_vhq"
+        )
+        sr = 44100
+
     orig_audio = track_audio.copy()
 
-    if sr != 44100:
-        raise ValueError("Sample rate should be 44100")
     augmented_gain = None
 
     if args.loudnorm_input_lufs:  # If you want to use loud-normalized input
@@ -169,6 +176,8 @@ def main(input, mix_coefficient):
         args, our_model, device, track_audio, track_name, meter, augmented_gain
     )
 
+    if orig_sr == 44100:
+        orig_audio = int16_to_float32(orig_audio)
     orig_audio = int16_to_float32(orig_audio)
     track_lufs = meter.integrated_loudness(orig_audio.T)
     augmented_gain = args.save_output_loudnorm - track_lufs
@@ -206,11 +215,16 @@ with gr.Blocks() as demo:
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%">
-                A demo for "Music De-limiter via Sample-wise Gain Inversion" to appear in WASPAA 2023.
-                You can first upload a music (.wav or .mp3, 44.1kHz) file and then press "De-limit" button to apply the De-limiter. Since we use a CPU instead of a GPU, it may require a few minute.
-                Then, you can apply a Parallel Mix technique, which is a simple linear mixing technique of "loudness normalized input" and the "de-limiter output".
-                You can modify the mixing coefficient by yourself.
-                If the coefficient is 0.3 then the output will be "loudness_normalized_input * 0.3 + de-limiter_output * 0.7"
+                A demo for "Music De-limiter via Sample-wise Gain Inversion" to appear in WASPAA 2023.<br>
+                Upload a stereo music (tested with .wav, .mp3, .m4a) file and then press "De-limit" button to apply the De-limiter.<br>
+                The processing is based on 44.1kHz sample rate. Other sample rate will be automatically resampled to 44.1kHz.<br>
+                Since we use a CPU instead of a GPU, it may require a few seconds to minutes.<br>
+                Then, you can apply a Parallel Mix technique, which is a simple linear mixing technique of "loudness normalized input" and the "de-limiter output", similar to Parallel Compression.<br>
+                If the coefficient is 0.3 then the output will be the "loudness_normalized_input * 0.3 + de-limiter_output * 0.7"<br>
+                Check our Paper <a href="https://arxiv.org/abs/2308.01187">[arXiv]</a>
+                Codes <a href="<a href="https://github.com/jeonchangbin49/De-limiter">[GitHub]</a>
+                Audio samples <a href="<a href="https://catnip-leaf-c6a.notion.site/Music-De-limiter-7072c0e725fd42249ff78cbbaedc95d7?pvs=4">[Notion]</a> <br>
+                Please let me know any issues or comments on [email protected] or the "Community" page (the upper right section of this page).
             </div>
         """
     )