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

update notation

Browse files
Files changed (1) hide show
  1. pdf_processor.py +1 -1
pdf_processor.py CHANGED
@@ -238,7 +238,7 @@ class PDFProcessor:
238
  final_summary = summary_chain.invoke({"texts": "\n".join([doc.page_content for doc in documents]), "language": self.language})
239
  return final_summary
240
 
241
- # Get the top k documents by score
242
  def get_summary(self, vectorstore, ai_model, type_model, api_key, project_id_watsonx, just_get_documments=False, k=10):
243
 
244
  final_summary_prompt = PromptTemplate(
 
238
  final_summary = summary_chain.invoke({"texts": "\n".join([doc.page_content for doc in documents]), "language": self.language})
239
  return final_summary
240
 
241
+ @spaces.GPU
242
  def get_summary(self, vectorstore, ai_model, type_model, api_key, project_id_watsonx, just_get_documments=False, k=10):
243
 
244
  final_summary_prompt = PromptTemplate(