uijnbh commited on
Commit
7492ba4
1 Parent(s): fdeee21

Parameterise index name

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,7 +8,6 @@ import openai
8
  import textwrap
9
 
10
  MODEL_NAME = 'sentence-transformers/all-MiniLM-L6-v2'
11
- INDEX_NAME = 'charity-assistant'
12
 
13
  def text_to_embedding(text):
14
  inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
@@ -52,7 +51,7 @@ def recommend(user_query: str) -> str:
52
  return motivation
53
 
54
  pc=Pinecone()
55
- index = pc.Index(INDEX_NAME)
56
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
57
  model = AutoModel.from_pretrained(MODEL_NAME)
58
  openai.api_key = os.environ['OPENAI_API_KEY']
 
8
  import textwrap
9
 
10
  MODEL_NAME = 'sentence-transformers/all-MiniLM-L6-v2'
 
11
 
12
  def text_to_embedding(text):
13
  inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
 
51
  return motivation
52
 
53
  pc=Pinecone()
54
+ index = pc.Index(os.environ['INDEX_NAME'])
55
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
56
  model = AutoModel.from_pretrained(MODEL_NAME)
57
  openai.api_key = os.environ['OPENAI_API_KEY']