Update app.py

app.py

@@ -14,6 +14,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 import PyPDF2
 import traceback
 import os
+pip install spaces
+from spaces import GPU
 
 import shutil
 from pathlib import Path
@@ -50,6 +52,7 @@ model = None
 tokenizer = None
 generation_config = None
 
+@spaces.GPU
 def test_llm_generation():
     try:
         test_prompt = "Hello, how are you today?"
@@ -67,7 +70,7 @@ def test_llm_generation():
     except Exception as e:
         add_log(f"❌ LLM quick test failed: {e}")
 
-
+@spaces.GPU
 def initialize_model():
     global model, tokenizer, generation_config
 
@@ -167,7 +170,7 @@ class PodcastGenerator:
             error_msg = f"❌ PDF extraction failed: {str(e)}"
             add_log(error_msg)
             raise Exception(error_msg)
-
+    @spaces.GPU
     async def postprocess_conversation(self, raw_text: str) -> str:
         """Run LLM again to enforce strict Speaker 1/2 format"""
         prompt = f"""
@@ -345,7 +348,8 @@ Now format the following:
         
         add_log(f"✅ Fallback podcast created with {len(podcast_lines)} lines")
         return result
-
+        
+    @spaces.GPU
     async def generate_script(self, prompt: str, language: str, file_obj=None, progress=None) -> Dict:
         """Improved script generation with better error handling"""
         if not model_loaded or not self.model or not self.tokenizer: