Spaces:
Runtime error
Runtime error
modified filea
Browse files
app.py
CHANGED
|
@@ -26,12 +26,13 @@ NEO_DB_PASSWORD = os.getenv('NEO_DB_PASSWORD')
|
|
| 26 |
|
| 27 |
|
| 28 |
# MongoDB setup
|
| 29 |
-
mongo_client = MongoClient(
|
| 30 |
db = mongo_client["pdf_db"]
|
| 31 |
chunks_collection = db["chunks"]
|
| 32 |
|
| 33 |
# Neo4j setup
|
| 34 |
-
neo4j_driver = GraphDatabase.driver(
|
|
|
|
| 35 |
|
| 36 |
# Load pre-trained BERT model and tokenizer
|
| 37 |
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
|
|
|
|
| 26 |
|
| 27 |
|
| 28 |
# MongoDB setup
|
| 29 |
+
mongo_client = MongoClient(MONGO_DB_URL)
|
| 30 |
db = mongo_client["pdf_db"]
|
| 31 |
chunks_collection = db["chunks"]
|
| 32 |
|
| 33 |
# Neo4j setup
|
| 34 |
+
neo4j_driver = GraphDatabase.driver(NEO_DB_HOST, auth=(NEO_DB_USER, NEO_DB_PASSWORD))
|
| 35 |
+
|
| 36 |
|
| 37 |
# Load pre-trained BERT model and tokenizer
|
| 38 |
tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
|