Update app.py

app.py

@@ -1,22 +1,27 @@
 import gradio as gr
-from voicefixer import VoiceFixer
 import os
 import torch
 
-# --- 1. Setup Logic ---
-device = "cuda" if torch.cuda.is_available() else "cpu"
-print(f"Using device: {device}")
+# --- 1. Setup Logic (Lazy Loading) ---
+# Hum model ko globally load nahi karenge taaki app turant start ho jaye
+vf_model = None
 
-# Initialize VoiceFixer
-try:
-    vf = VoiceFixer()
-except Exception as e:
-    print(f"Error loading VoiceFixer: {e}")
+def get_model():
+    global vf_model
+    if vf_model is None:
+        print("Loading VoiceFixer Model...")
+        from voicefixer import VoiceFixer
+        vf_model = VoiceFixer()
+    return vf_model
 
 def fix_audio_ts(audio_path):
     if audio_path is None:
         return None
     
+    # Model load karo (agar pehle se loaded nahi hai)
+    vf = get_model()
+    
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     output_file = "ts_studio_enhanced.wav"
     
     # Mode 0: Cleaning + Restoration