Update app.py

app.py

@@ -485,6 +485,7 @@ def main():
                 print("Model initialized.")
                 
                 query_embedding = model.encode([query])
+                query_embedding = np.array(query_embedding).reshape(1, -1).astype('float32')
                 # Retrieve examples by title similarity (or abstract, depending on your preference)
                 scores_title, retrieved_examples_title = st.session_state.dataclysm_title_indexed.get_nearest_examples('title_embedding', query_embedding, k=top_k)
                 df_query = pd.DataFrame(retrieved_examples_title)