Update app.py

app.py

@@ -100,7 +100,6 @@ def prepare_images_and_masks(total_mask, original_image, add_color_image, add_ed
     return add_color_image_tensor, original_image_tensor, total_mask, add_edge_mask, remove_edge_mask
 
 def guess_prompt_handler(original_image, add_color_image, add_edge_image):
-    # 将张量转换为 NumPy 数组
     original_image_tensor = load_and_preprocess_image(original_image)
     
     if add_color_image:
@@ -114,33 +113,27 @@ def guess_prompt_handler(original_image, add_color_image, add_edge_image):
     original_image_numpy = tensor_to_numpy(original_image_tensor.squeeze(0))
     add_color_image_numpy = tensor_to_numpy(add_color_image_tensor.squeeze(0))
     add_edge_mask_numpy = tensor_to_numpy(add_edge_mask.squeeze(0).unsqueeze(-1))
-
     original_image_numpy = cv2.cvtColor(original_image_numpy, cv2.COLOR_RGB2BGR)
     add_color_image_numpy = cv2.cvtColor(add_color_image_numpy, cv2.COLOR_RGB2BGR)
-    
-    # 创建临时文件，保存 NumPy 数组为图像文件
+
     original_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
     add_color_image_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
     add_edge_mask_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png", mode='w+b')
     
-    # 保存 NumPy 数组为 PNG 图像文件
     cv2.imwrite(original_image_file.name, original_image_numpy)
     cv2.imwrite(add_color_image_file.name, add_color_image_numpy)
     cv2.imwrite(add_edge_mask_file.name, add_edge_mask_numpy)
-    
-    # 确保文件关闭以保证内容写入磁盘
+
     original_image_file.close()
     add_color_image_file.close()
     add_edge_mask_file.close()
     
-    # 调用 API，传递临时文件的路径
     res = client.predict(
         handle_file(original_image_file.name),
         handle_file(add_color_image_file.name),
         handle_file(add_edge_mask_file.name)
     )
     
-    # 删除临时文件，确保它们存在且已被创建
     if original_image_file and os.path.exists(original_image_file.name):
         os.remove(original_image_file.name)
     if add_color_image_file and os.path.exists(add_color_image_file.name):