Update core.py

core.py

@@ -4,6 +4,10 @@ import json
 import argparse
 import subprocess
 
+# اضافه کردن پشتیبانی از GPU و ZeroGPU
+import spaces
+import torch
+
 now_dir = os.getcwd()
 sys.path.append(now_dir)
 
@@ -37,6 +41,7 @@ locales = list({voice["Locale"] for voice in voices_data})
 
 
 # Infer
+@spaces.GPU
 def run_infer_script(
     f0up_key,
     filter_radius,
@@ -55,6 +60,11 @@ def run_infer_script(
     clean_strength,
     export_format,
 ):
+    # بررسی و مقداردهی اولیه GPU
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    print(f"✅ Running inference on {device}")
+
+    # اجرای پردازش
     infer_pipeline(
         f0up_key,
         filter_radius,
@@ -73,11 +83,13 @@ def run_infer_script(
         clean_strength,
         export_format,
     )
+
     return f"File {input_path} inferred successfully.", output_path.replace(
         ".wav", f".{export_format.lower()}"
     )
 
 
+
 # Batch infer
 def run_batch_infer_script(
     f0up_key,