stivenDR14
commited on
Commit
·
a7a404a
1
Parent(s):
c83b635
update decorators
Browse files- pdf_processor.py +2 -2
pdf_processor.py
CHANGED
@@ -92,14 +92,14 @@ def authenticate_watsonx(api_key):
|
|
92 |
print("Response:", response.text)
|
93 |
return None
|
94 |
|
95 |
-
|
96 |
class PDFProcessor:
|
97 |
def __init__(self):
|
98 |
self.language = "English"
|
99 |
|
100 |
def set_language(self, language):
|
101 |
self.language = language
|
102 |
-
|
103 |
def set_llm(self, ai_model, type_model, api_key, project_id_watsonx):
|
104 |
if ai_model == "Open AI / GPT-4o-mini":
|
105 |
current_llm = ChatOpenAI(
|
|
|
92 |
print("Response:", response.text)
|
93 |
return None
|
94 |
|
95 |
+
@spaces.GPU
|
96 |
class PDFProcessor:
|
97 |
def __init__(self):
|
98 |
self.language = "English"
|
99 |
|
100 |
def set_language(self, language):
|
101 |
self.language = language
|
102 |
+
@spaces.GPU
|
103 |
def set_llm(self, ai_model, type_model, api_key, project_id_watsonx):
|
104 |
if ai_model == "Open AI / GPT-4o-mini":
|
105 |
current_llm = ChatOpenAI(
|