Jonas Leeb
commited on
Commit
·
5e54614
1
Parent(s):
c8a50e8
plotting bug fix 2
Browse files
app.py
CHANGED
@@ -185,7 +185,7 @@ class ArxivSearch:
|
|
185 |
marker=dict(size=3.5, color='orange', opacity=0.75),
|
186 |
name='Results'
|
187 |
)
|
188 |
-
if
|
189 |
query_trace = go.Scatter3d(
|
190 |
x=query_point[:, 0],
|
191 |
y=query_point[:, 1],
|
@@ -284,6 +284,9 @@ class ArxivSearch:
|
|
284 |
|
285 |
|
286 |
def search_function(self, query, embedding):
|
|
|
|
|
|
|
287 |
# Load or switch embedding model here if needed
|
288 |
if embedding == "tfidf":
|
289 |
results = self.keyword_match_ranking(query)
|
|
|
185 |
marker=dict(size=3.5, color='orange', opacity=0.75),
|
186 |
name='Results'
|
187 |
)
|
188 |
+
if not "tfidf" and self.query_encoding is not None and self.query_encoding.shape[0] > 0:
|
189 |
query_trace = go.Scatter3d(
|
190 |
x=query_point[:, 0],
|
191 |
y=query_point[:, 1],
|
|
|
284 |
|
285 |
|
286 |
def search_function(self, query, embedding):
|
287 |
+
# Preprocess the query
|
288 |
+
query = query.strip().lower()
|
289 |
+
|
290 |
# Load or switch embedding model here if needed
|
291 |
if embedding == "tfidf":
|
292 |
results = self.keyword_match_ranking(query)
|