update notations in gradio ui

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from pdf_processor import PDFProcessor
 from utils import AI_MODELS, TRANSLATIONS
 
@@ -57,15 +58,19 @@ class PDFProcessorUI:
         else:
             return gr.update(visible=False), gr.update(visible=False)
     
+    @spaces.GPU
     def process_pdf(self, vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model, type_model, api_key, project_id_watsonx):
         return self.processor.process_pdf(vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model, type_model, api_key, project_id_watsonx)
     
+    @spaces.GPU
     def qa_interface(self, vectorstore, message, history, ai_model, type_model, api_key, project_id_watsonx):
         return self.processor.get_qa_response(vectorstore, message, history, ai_model, type_model, api_key, project_id_watsonx)
     
+    @spaces.GPU
     def summarize_interface(self, vectorstore, ai_model, type_model, api_key, project_id_watsonx):
         return self.processor.get_summary(vectorstore, ai_model, type_model, api_key, project_id_watsonx)
     
+    @spaces.GPU
     def specialist_opinion(self, vectorstore, ai_model, type_model, api_key, project_id_watsonx, specialist_prompt):
         return self.processor.get_specialist_opinion(vectorstore, ai_model, type_model, api_key, project_id_watsonx, specialist_prompt)
     

pdf_processor.py

@@ -1,5 +1,4 @@
 import json
-import tempfile
 import spaces
 from langchain_community.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -92,7 +91,6 @@ def authenticate_watsonx(api_key):
         print("Response:", response.text)
         return None
 
-@spaces.GPU
 class PDFProcessor:
     def __init__(self):
         self.language = "English"