ShawnAI commited on
Commit
99b9f14
·
1 Parent(s): 52901ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -16,10 +16,13 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
16
  #OPENAI_API_KEY = ""
17
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
18
  OPENAI_TEMP = 0
 
19
  PINECONE_KEY = os.environ["PINECONE_KEY"]
20
  PINECONE_ENV = "asia-northeast1-gcp"
21
  PINECONE_INDEX = "3gpp"
22
 
 
 
23
  # return top-k text chunk from vector store
24
  VECTOR_SEARCH_TOP_K = 10
25
 
@@ -57,7 +60,7 @@ def init_model(api_key):
57
  try:
58
  if api_key and api_key.startswith("sk-") and len(api_key) > 50:
59
 
60
- embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
61
 
62
  pinecone.init(api_key = PINECONE_KEY,
63
  environment = PINECONE_ENV)
 
16
  #OPENAI_API_KEY = ""
17
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
18
  OPENAI_TEMP = 0
19
+
20
  PINECONE_KEY = os.environ["PINECONE_KEY"]
21
  PINECONE_ENV = "asia-northeast1-gcp"
22
  PINECONE_INDEX = "3gpp"
23
 
24
+ EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
25
+
26
  # return top-k text chunk from vector store
27
  VECTOR_SEARCH_TOP_K = 10
28
 
 
60
  try:
61
  if api_key and api_key.startswith("sk-") and len(api_key) > 50:
62
 
63
+ embeddings = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
64
 
65
  pinecone.init(api_key = PINECONE_KEY,
66
  environment = PINECONE_ENV)