small fixes

app.py

@@ -21,15 +21,16 @@
 
 import logging
 import os
-from pydub import AudioSegment
 import tempfile
 import time
+import urllib.request
 from datetime import datetime
 
 import gradio as gr
 import torch
 import torchaudio
-import urllib.request
+from pydub import AudioSegment
+
 from separate import load_audio, load_model, separate
 
 

requirements.txt

@@ -2,6 +2,6 @@ https://download.pytorch.org/whl/cpu/torch-1.13.1%2Bcpu-cp38-cp38-linux_x86_64.w
 https://download.pytorch.org/whl/cpu/torchaudio-0.13.1%2Bcpu-cp38-cp38-linux_x86_64.whl
 
 numpy
-
+pydub
 huggingface_hub
 ffmpeg

separate.py

@@ -1,21 +1,16 @@
 #!/usr/bin/env python3
 # Copyright    2023  Xiaomi Corp.        (authors: Fangjun Kuang)
 
-# Please see ./run.sh for usage
-
+from functools import lru_cache
 from typing import Optional, Tuple
 
-from huggingface_hub import hf_hub_download
-
 import ffmpeg
 import numpy as np
 import torch
-import soundfile as sf
 import torchaudio
-from functools import lru_cache
+from huggingface_hub import hf_hub_download
 from pydub import AudioSegment
 
-
 from unet import UNet