Update app_main.py
Browse files- 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:
|