stivenDR14
commited on
Commit
·
77773f3
1
Parent(s):
629027d
delete spaces.gpu
Browse files- 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 |
-
|
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
|