Update inference.py

inference.py

@@ -13,6 +13,7 @@ import soundfile as sf
 import torch.nn as nn
 import numpy as np
 from assets.i18n.i18n import I18nAuto
+import spaces
 
 # Colab kontrolü
 try:
@@ -138,6 +139,7 @@ def run_folder(model, args, config, device, verbose: bool = False):
 
     print(i18n("elapsed_time").format(time.time() - start_time))
 
+@spaces.GPU
 def proc_folder(args):
     parser = argparse.ArgumentParser(description=i18n("proc_folder_description"))
     parser.add_argument("--model_type", type=str, default='mdx23c', help=i18n("model_type_help"))
@@ -171,7 +173,7 @@ def proc_folder(args):
         print(i18n("cuda_available"))
         device = f'cuda:{args.device_ids[0]}' if type(args.device_ids) == list else f'cuda:{args.device_ids}'
     elif torch.backends.mps.is_available():
-         device = "mps"
+        device = "mps"
 
     print(i18n("using_device").format(device))