Update app.py

app.py

@@ -61,7 +61,8 @@ def embed_question(question, model, tokenizer):
         with torch.no_grad():
             outputs = model(**inputs)
         embedding = outputs.last_hidden_state.mean(dim=1).cpu().numpy().astype(np.float32)
-        logging.debug(f"Question embedding: {embedding}")
+        logging.debug(f"Question embedding shape: {embedding.shape}")
+        logging.debug(f"Question embedding content: {embedding}")
         return embedding
     except Exception as e:
         logging.error(f"Error embedding question: {e}")
@@ -77,6 +78,10 @@ def retrieve_and_generate(question):
 
         # Search in FAISS index
         try:
+            if question_embedding.ndim == 2 and question_embedding.shape[0] == 1:
+                logging.debug("Correct question embedding shape for FAISS search.")
+            else:
+                logging.error(f"Incorrect question embedding shape: {question_embedding.shape}")
             _, indices = gpu_index.search(question_embedding, k=1)
             logging.debug(f"Indices found: {indices}")
         except Exception as e: