Resolve search crash due to wrong shape

app.py

@@ -28,12 +28,12 @@ model = SentenceTransformer(
 def search(query, top_k: int = 10, rerank_multiplier: int = 4):
     # 1. Embed the query as float32
     start_time = time.time()
-    query_embedding = model.encode([query])
+    query_embedding = model.encode(query)
     embed_time = time.time() - start_time
 
     # 2. Quantize the query to ubinary
     start_time = time.time()
-    query_embedding_ubinary = quantize_embeddings(query_embedding, "ubinary")
+    query_embedding_ubinary = quantize_embeddings(query_embedding.reshape(1, -1), "ubinary")
     quantize_time = time.time() - start_time
 
     # 3. Search the binary index