uijnbh commited on
Commit
fdeee21
1 Parent(s): 4be9773
Files changed (1) hide show
  1. app.py +2 -2
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()