Update app.py

app.py

@@ -9,7 +9,9 @@ from audio_separator.separator import Separator
 device = "cuda" if torch.cuda.is_available() else "cpu"
 use_autocast = device == "cuda"
 
-# Model lists
+#=========================#
+#     Roformer Models     #
+#=========================#
 ROFORMER_MODELS = {
     'BS-Roformer-De-Reverb': 'deverb_bs_roformer_8_384dim_10depth.ckpt',
     'BS-Roformer-Viperx-1053': 'model_bs_roformer_ep_937_sdr_10.5309.ckpt',
@@ -25,11 +27,17 @@ ROFORMER_MODELS = {
     'MelBand Roformer Kim | InstVoc Duality V1 by Unwa': 'melband_roformer_instvoc_duality_v1.ckpt',
     'MelBand Roformer Kim | InstVoc Duality V2 by Unwa': 'melband_roformer_instvox_duality_v2.ckpt',
 }
+#=========================#
+#      MDX23C Models      #
+#=========================#
 MDX23C_MODELS = [
     'MDX23C-8KFFT-InstVoc_HQ.ckpt',
     'MDX23C-8KFFT-InstVoc_HQ_2.ckpt',
     'MDX23C_D1581.ckpt',
 ]
+#=========================#
+#     MDXN-NET Models     #
+#=========================#
 MDXNET_MODELS = [
     'UVR-MDX-NET-Crowd_HQ_1.onnx',
     'UVR-MDX-NET-Inst_1.onnx',
@@ -70,6 +78,9 @@ MDXNET_MODELS = [
     'Kim_Vocal_2.onnx',
     'Reverb_HQ_By_FoxJoy.onnx',
 ]
+#========================#
+#     VR-ARCH Models     #
+#========================#
 VR_ARCH_MODELS = [
     '1_HP-UVR.pth',
     '2_HP-UVR.pth',
@@ -99,6 +110,9 @@ VR_ARCH_MODELS = [
     'UVR-DeNoise-Lite.pth',
     'UVR-DeNoise.pth',
 ]
+#=======================#
+#     DEMUCS Models     #
+#=======================#
 DEMUCS_MODELS = [
     'hdemucs_mmi.yaml',
     'htdemucs.yaml',