Adventure123 commited on
Commit
2bcdd30
·
verified ·
1 Parent(s): ee5e861

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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['NEO4J_URI'],
32
- username=os.environ['NEO4J_USERNAME'],
33
- password=os.environ['NEO4J_PASSWORD'],
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'],