Update app.py

app.py

@@ -12,6 +12,7 @@ import threading
 import queue
 import os
 import logging
+import spaces
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -31,6 +32,7 @@ print(f"Using device: {device}")
 model = None
 tokenizer = None
 
+@spaces.GPU()
 def load_model():
     global model, tokenizer
     
@@ -137,6 +139,7 @@ def generate_podcast_script(api_key, content, duration, num_hosts):
     clean_text = re.sub(r'[^a-zA-Z0-9\s.,?!<>]', '', response.text)
     return clean_text
 
+@spaces.GPU()
 def text_to_speech(text, voice):
     global model, tokenizer
     if tokenizer is None or model is None:
@@ -153,6 +156,7 @@ def mel_to_audio(mel):
     # Placeholder implementation
     return np.zeros(24000, dtype=np.float32)  # 1 second of silence
 
+@spaces.GPU()
 def process_audio_segment(line, voice, result_queue):
     try:
         audio = text_to_speech(line, voice)
@@ -161,6 +165,7 @@ def process_audio_segment(line, voice, result_queue):
         logger.error(f"Error processing audio segment: {str(e)}")
         result_queue.put(None)
 
+@spaces.GPU()
 def render_podcast(api_key, script, voice1, voice2, num_hosts):
     lines = [line for line in script.split('\n') if line.strip()]
     audio_segments = []