Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,8 @@ from langchain_community.vectorstores import Neo4jVector
|
|
5 |
from transformers import AutoTokenizer, AutoModel
|
6 |
import torch
|
7 |
|
|
|
|
|
8 |
# Custom Embedding Class
|
9 |
class CustomHuggingFaceEmbeddings:
|
10 |
def __init__(self, model_name="sentence-transformers/all-MiniLM-L6-v2"):
|
@@ -28,9 +30,9 @@ class CustomHuggingFaceEmbeddings:
|
|
28 |
def setup_vector_index():
|
29 |
return Neo4jVector.from_existing_graph(
|
30 |
CustomHuggingFaceEmbeddings(),
|
31 |
-
url=os.environ
|
32 |
-
username=os.environ
|
33 |
-
password=os.environ
|
34 |
index_name='articles',
|
35 |
node_label="Article",
|
36 |
text_node_properties=['topic', 'title', 'abstract'],
|
|
|
5 |
from transformers import AutoTokenizer, AutoModel
|
6 |
import torch
|
7 |
|
8 |
+
print(f"Username Neo4j: {os.environ.get('NEO4J_USERNAME')}")
|
9 |
+
|
10 |
# Custom Embedding Class
|
11 |
class CustomHuggingFaceEmbeddings:
|
12 |
def __init__(self, model_name="sentence-transformers/all-MiniLM-L6-v2"):
|
|
|
30 |
def setup_vector_index():
|
31 |
return Neo4jVector.from_existing_graph(
|
32 |
CustomHuggingFaceEmbeddings(),
|
33 |
+
url=os.environ.get('NEO4J_URI'),
|
34 |
+
username=os.environ.get('NEO4J_USERNAME'),
|
35 |
+
password=os.environ.get('NEO4J_PASSWORD'),
|
36 |
index_name='articles',
|
37 |
node_label="Article",
|
38 |
text_node_properties=['topic', 'title', 'abstract'],
|