stivenDR14 commited on
Commit
a7a404a
·
1 Parent(s): c83b635

update decorators

Browse files
Files changed (1) hide show
  1. 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(