From 0ab13cbfe1a6f6bcf5fa6ff4dac7f0dd6ae48ded Mon Sep 17 00:00:00 2001
From: shenyuhan <shenyuhan@baidu.com>
Date: Mon, 4 Nov 2019 22:01:50 +0800
Subject: [PATCH] 1.add config batch_size; 2.delete req_id_ for every file

---
 paddlehub/serving/app_single.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/paddlehub/serving/app_single.py b/paddlehub/serving/app_single.py
index 6ef2378a..7a5a02e6 100644
--- a/paddlehub/serving/app_single.py
+++ b/paddlehub/serving/app_single.py
@@ -225,6 +225,7 @@ def predict_semantic_segmentation(module, input_img, id, batch_size):
     for index in range(len(input_img)):
         # special
         item = input_img[index]
+        output_file_path = ""
         with open(results[index]["processed"], "rb") as fp:
             # special
             b_head = "data:image/png;base64"
@@ -234,12 +235,14 @@ def predict_semantic_segmentation(module, input_img, id, batch_size):
             base64_list.append(b_img)
             results[index]["origin"] = results[index]["origin"].replace(
                 id + "_", "")
+            output_file_path = results[index]["processed"]
             results[index]["processed"] = results[index]["processed"].replace(
                 id + "_", "")
             results[index].update({"base64": b_img})
             results_pack.append(results[index])
         os.remove(item)
-        os.remove(results[index]["processed"])
+        if output_file_path != "":
+            os.remove(output_file_path)
     return results_pack
 
 
-- 
GitLab