bhlewis commited on
Commit
e0eddce
1 Parent(s): 9f01b33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -54,7 +54,7 @@ index = faiss.IndexFlatIP(embeddings.shape[1])
54
  index.add(embeddings)
55
 
56
  # Load BERT model for encoding search queries
57
- model = SentenceTransformer('all-mpnet-base-v2')
58
 
59
  # Create TF-IDF vectorizer
60
  tfidf_vectorizer = TfidfVectorizer(stop_words='english')
@@ -139,4 +139,4 @@ iface = gr.Interface(
139
  )
140
 
141
  if __name__ == "__main__":
142
- iface.launch()
 
54
  index.add(embeddings)
55
 
56
  # Load BERT model for encoding search queries
57
+ model = SentenceTransformer('anferico/bert-for-patents')
58
 
59
  # Create TF-IDF vectorizer
60
  tfidf_vectorizer = TfidfVectorizer(stop_words='english')
 
139
  )
140
 
141
  if __name__ == "__main__":
142
+ iface.launch()