Update app.py

app.py

@@ -133,27 +133,26 @@ def run_cmd(command):
 
 def run(image):
     
-    if os.path.isdir(f"Temp{uid}"):
-        shutil.rmtree(f"Temp{uid}")
+    if os.path.isdir(f"Temp"):
+        shutil.rmtree(f"Temp")
     
-    os.makedirs(f"Temp{uid}")
-    os.makedirs(f"Temp{uid}/input")
-    os.makedirs(f"Temp{uid}/final_output")
+    os.makedirs(f"Temp")
+    os.makedirs(f"Temp/input")
 
     print(type(image))
-    cv2.imwrite(f"Temp{uid}/input/input_img.png", image)
+    cv2.imwrite(f"Temp/input/input_img.png", image)
 
     command = ("python run.py --input_folder "
-            + f"Temp{uid}/input"
+            + f"Temp/input"
             + " --output_folder "
-            + f"Temp{uid}"
+            + f"Temp"
             + " --GPU "
             + "-1"
             + " --with_scratch")
     run_cmd(command)
 
-    result_restoration = Image.open(f"Temp{uid}/final_output/input_img.png")
-    shutil.rmtree(f"Temp{uid}")
+    result_restoration = Image.open(f"Temp/final_output/input_img.png")
+    shutil.rmtree(f"Temp")
     
     result_colorization = inferColorization(result_restoration)