Update app.py

app.py

@@ -18,7 +18,11 @@ def encode_text(text):
 def find_best_response(user_input, response_pool):
     user_embedding = encode_text(user_input)
     response_embeddings = np.array([encode_text(resp) for resp in response_pool])
-    # Check if response_embeddings need reshaping
+    
+    # Ensure response_embeddings are 2D
+    if response_embeddings.ndim == 3:
+        response_embeddings = response_embeddings.squeeze(axis=1)
+    
     similarities = cosine_similarity(user_embedding, response_embeddings).flatten()
     best_response_index = np.argmax(similarities)
     return response_pool[best_response_index]
@@ -48,3 +52,4 @@ iface = gr.Interface(
 # Launch the interface
 iface.launch()
 
+