prthm11 commited on
Commit
4f59510
·
verified ·
1 Parent(s): 46bbf3c

Update app_main.py

Browse files
Files changed (1) hide show
  1. app_main.py +4 -1
app_main.py CHANGED
@@ -372,7 +372,10 @@ def similarity_matching(input_json_path: str) -> str:
372
 
373
  img_matrix = np.array([img["embeddings"] for img in embedding_json])
374
  sprite_matrix = np.array(sprite_features)
375
-
 
 
 
376
  try:
377
  similarity = np.matmul(sprite_matrix, img_matrix.T)
378
  except ValueError as ve:
 
372
 
373
  img_matrix = np.array([img["embeddings"] for img in embedding_json])
374
  sprite_matrix = np.array(sprite_features)
375
+
376
+ if sprite_matrix.size == 0 or img_matrix.size == 0:
377
+ raise RuntimeError("❌ No valid embeddings found for sprites or reference images.")
378
+
379
  try:
380
  similarity = np.matmul(sprite_matrix, img_matrix.T)
381
  except ValueError as ve: