Update app.py

app.py

@@ -134,6 +134,12 @@ def recommend_products_for_user(user_id, top_n=1):
 
         product_embeddings = np.array(product_embeddings)
 
+        # 차원 확인 및 조정
+        if product_embeddings.ndim == 3:  # 3D 배열인 경우 평균 풀링 적용
+            product_embeddings = product_embeddings.mean(axis=1)
+        elif product_embeddings.ndim == 1:  # 1D 배열인 경우 2D로 변환
+            product_embeddings = product_embeddings.reshape(1, -1)
+
         # Cosine Similarity 계산
         similarities = cosine_similarity(user_embedding, product_embeddings).flatten()