Update app.py

app.py

@@ -6,6 +6,7 @@ from generate_transcript import TranscriptProcessor
 from generate_audio import TTSGenerator
 import pickle
 import os
+import space
 
 # Define paths
 pdf_path = './resources/uploaded_pdf.pdf'
@@ -16,6 +17,7 @@ audio_output_path = 'final_podcast_audio.mp3'
 
 
 # Combined function to perform all steps sequentially
+@space.GPU
 def process_pdf_to_podcast(pdf_file):
     # Step 1: Extract Text from PDF
     with open(pdf_path, 'wb') as f:
@@ -52,6 +54,7 @@ def process_pdf_to_podcast(pdf_file):
 
 
 # Final Step: Generate Audio after optional adjustments
+@space.GPU
 def generate_audio_from_modified_text(tts_ready_text):
     # Save any modified TTS-ready transcript
     with open(tts_ready_path, 'wb') as f: