stivenDR14 commited on
Commit
77773f3
·
1 Parent(s): 629027d

delete spaces.gpu

Browse files
Files changed (1) hide show
  1. pdf_processor.py +1 -2
pdf_processor.py CHANGED
@@ -97,7 +97,7 @@ class PDFProcessor:
97
 
98
  def set_language(self, language):
99
  self.language = language
100
- @spaces.GPU
101
  def set_llm(self, ai_model, type_model, api_key, project_id_watsonx):
102
  if ai_model == "Open AI / GPT-4o-mini":
103
  current_llm = ChatOpenAI(
@@ -141,7 +141,6 @@ class PDFProcessor:
141
  )
142
  return current_llm, embeding_model
143
 
144
- @spaces.GPU
145
  def process_pdf(self, vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model, type_model, api_key, project_id_watsonx):
146
  defined_chunk_size = 1000
147
  defined_chunk_overlap = 150
 
97
 
98
  def set_language(self, language):
99
  self.language = language
100
+
101
  def set_llm(self, ai_model, type_model, api_key, project_id_watsonx):
102
  if ai_model == "Open AI / GPT-4o-mini":
103
  current_llm = ChatOpenAI(
 
141
  )
142
  return current_llm, embeding_model
143
 
 
144
  def process_pdf(self, vectorstore, pdf_file, chunk_size, chunk_overlap, ai_model, type_model, api_key, project_id_watsonx):
145
  defined_chunk_size = 1000
146
  defined_chunk_overlap = 150