nkcong206 commited on
Commit
0bd4b96
·
1 Parent(s): ee39452
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -318,7 +318,8 @@ def format_docs(docs):
318
  return "\n\n".join(doc.page_content for doc in docs)
319
 
320
  @st.cache_resource
321
- def compute_rag_chain(model, embd, docs_texts):
 
322
  results = recursive_embed_cluster_summarize(model, embd, docs_texts, level=1, n_levels=3)
323
  all_texts = docs_texts.copy()
324
  for level in sorted(results.keys()):
 
318
  return "\n\n".join(doc.page_content for doc in docs)
319
 
320
  @st.cache_resource
321
+ def compute_rag_chain(_model, embd, docs_texts):
322
+ model = _model
323
  results = recursive_embed_cluster_summarize(model, embd, docs_texts, level=1, n_levels=3)
324
  all_texts = docs_texts.copy()
325
  for level in sorted(results.keys()):