ArthurSrz commited on
Commit
b55d261
Β·
1 Parent(s): 5b582e0

Restore: Full nano-graphrag dependencies (graspologic, scipy, umap-learn) now that HF Space has CPU upgrade with sufficient resources

Browse files
Files changed (2) hide show
  1. app.py +1 -4
  2. requirements.txt +4 -1
app.py CHANGED
@@ -42,15 +42,12 @@ class BorgesGraphRAG:
42
  try:
43
  print(f"πŸ”„ Loading GraphRAG instance for {book_folder}...")
44
  if book_folder not in self.instances:
45
- # Try with reduced functionality to avoid graspologic issues
46
  self.instances[book_folder] = GraphRAG(
47
  working_dir=book_folder,
48
  best_model_func=gpt_4o_mini_complete,
49
  cheap_model_func=gpt_4o_mini_complete,
50
  best_model_max_async=3,
51
- cheap_model_max_async=3,
52
- # Use basic embedding model to avoid graspologic
53
- enable_naive_rag=True
54
  )
55
  print(f"βœ… GraphRAG instance created for {book_folder}")
56
  else:
 
42
  try:
43
  print(f"πŸ”„ Loading GraphRAG instance for {book_folder}...")
44
  if book_folder not in self.instances:
 
45
  self.instances[book_folder] = GraphRAG(
46
  working_dir=book_folder,
47
  best_model_func=gpt_4o_mini_complete,
48
  cheap_model_func=gpt_4o_mini_complete,
49
  best_model_max_async=3,
50
+ cheap_model_max_async=3
 
 
51
  )
52
  print(f"βœ… GraphRAG instance created for {book_folder}")
53
  else:
requirements.txt CHANGED
@@ -10,4 +10,7 @@ aiohttp>=3.8.0
10
  python-dotenv
11
  pandas
12
  scikit-learn
13
- sentence-transformers
 
 
 
 
10
  python-dotenv
11
  pandas
12
  scikit-learn
13
+ sentence-transformers
14
+ scipy
15
+ graspologic
16
+ umap-learn