Update extract_text_from_pdf.py

extract_text_from_pdf.py

@@ -2,6 +2,7 @@
 
 import os
 import torch
+import spaces
 from PyPDF2 import PdfReader
 from accelerate import Accelerator
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -9,6 +10,7 @@ from tqdm import tqdm
 import warnings
 
 
+
 warnings.filterwarnings('ignore')
 
 
@@ -16,7 +18,7 @@ class PDFTextExtractor:
     """
     A class to handle PDF text extraction and preprocessing for podcast preparation.
     """
-
+    @spaces.GPU
     def __init__(self, pdf_path, output_path, model_name="meta-llama/Llama-3.2-1B-Instruct"):
         """
         Initialize the PDFTextExtractor with paths and model details.
@@ -109,6 +111,7 @@ class PDFTextExtractor:
         
         return chunks
 
+    @spaces.GPU
     def process_chunk(self, text_chunk):
         """Process a text chunk with the model and return the cleaned text."""
         conversation = [