Update to denoisers 0.2.0

app.py

@@ -1,9 +1,6 @@
 """Gradio demo for denoisers."""
-import tempfile
-from pathlib import Path
 
 import gradio as gr
-import numpy as np
 import torch
 import torchaudio
 from denoisers import UNet1DModel, WaveUNetModel
@@ -16,7 +13,7 @@ MODELS = [
 ]
 
 
-def denoise(model_name: str, audio_path: str):
+def denoise(model_name: str, audio_path: str) -> str:
     """Denoise audio."""
     if "unet1d" in model_name:
         model = UNet1DModel.from_pretrained(model_name)

requirements.txt

@@ -3,7 +3,7 @@ torch
 torchaudio
 pytorch-lightning
 pedalboard
-denoisers==0.1.7
+denoisers==0.2.0
 transformers
 librosa
 wandb