Spaces:
Runtime error
Runtime error
Restore: Full nano-graphrag dependencies (graspologic, scipy, umap-learn) now that HF Space has CPU upgrade with sufficient resources
Browse files- app.py +1 -4
- 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
|