Update app.py

app.py

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