Swapping to zero-gpu

Previous commit works on paid hardware eg. L4s

web_ui.py

@@ -78,7 +78,9 @@ def validate_audio_input(audio_path):
     except Exception as e:
         raise ValueError(f"Audio processing failed: {e}")
 
-@spaces.GPU
+#@spaces.GPU
+@spaces.GPU(duration=120) # Swapping to zero-gpu
+
 def generate_speech(text, prompt_audio_path):
     """Generate speech with GPU acceleration for HF Spaces."""
     try:
@@ -243,12 +245,14 @@ def build_interface():
     return demo
 
 if __name__ == "__main__":
+
+    # Removing this for zero-gpu to work.
     # Initialize model at startup
-    try:
-        initialize_model()
-        logging.info("Model pre-loaded successfully")
-    except Exception as e:
-        logging.error(f"Failed to pre-load model: {e}")
+    #try:
+    #    initialize_model()
+    #    logging.info("Model pre-loaded successfully")
+    #except Exception as e:
+    #    logging.error(f"Failed to pre-load model: {e}")
     
     # Launch interface
     demo = build_interface()