Spaces:
Running
Running
Refactor
Browse files
app.py
CHANGED
@@ -9,8 +9,6 @@ import textwrap
|
|
9 |
|
10 |
MODEL_NAME = 'sentence-transformers/all-MiniLM-L6-v2'
|
11 |
INDEX_NAME = 'charity-assistant'
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
13 |
-
model = AutoModel.from_pretrained(MODEL_NAME)
|
14 |
|
15 |
def text_to_embedding(text):
|
16 |
inputs = tokenizer(text, padding=True, truncation=True, return_tensors="pt")
|
@@ -55,6 +53,8 @@ def recommend(user_query: str) -> str:
|
|
55 |
|
56 |
pc=Pinecone()
|
57 |
index = pc.Index(INDEX_NAME)
|
|
|
|
|
58 |
openai.api_key = os.environ['OPENAI_API_KEY']
|
59 |
recommender = gr.Interface(fn=recommend, inputs="text", outputs="text")
|
60 |
recommender.launch()
|
|
|
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")
|
|
|
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']
|
59 |
recommender = gr.Interface(fn=recommend, inputs="text", outputs="text")
|
60 |
recommender.launch()
|