Update api_server.py

api_server.py

@@ -165,8 +165,6 @@ def predict():
                 print(f"警告：{element} 沒有找到相關的 JPG 檔案")
                 continue
 
-            element_list.append(element)
-            
             for yolo_img in yolo_file: # 每張切圖yolo_img
                 print("***** 4. START CLIP *****")
                 client = Client(GRADIO_URL)
@@ -177,6 +175,7 @@ def predict():
                 )
                 top_k_words.append(clip_result) # CLIP預測3個結果(top_k_words)
                 encoded_images.append(image_to_base64(yolo_img))
+                element_list.append(element)
                 print(f"===== CLIP result:{top_k_words} =====\n")
 
             # if element_counts[element] > 1: #某隻角色的數量>1
@@ -196,15 +195,13 @@ def predict():
             'objects': [
                 {
                     'element': element,
-                    'images': [
+                    'images': 
                         {
                             'encoded_image': encoded_image,
                             'description_list': description_list
                         }                
-                        for encoded_image, description_list in zip(encoded_images, top_k_words)
-                    ]
                 }
-                for element, encoded_images, top_k_words in zip(element_list, encoded_images, top_k_words)
+                for element, encoded_image, description_list in zip(element_list, encoded_images, top_k_words)
             ]
         }