Update app.py
Browse files
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('
|
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()
|