Spaces:
Running
Running
Parameterise index name
Browse files
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']
|