Update app.py

app.py

@@ -10,12 +10,6 @@ import os
 
 
 
-# Empty the images folder before starting
-for path in os.listdir('images'):
-    os.remove(path)
-
-
-
 # Define global constants
 MAX_WIDTH = 700
 
@@ -69,10 +63,10 @@ if scale:
     scale_np = cv2.imdecode(scale_np, 1)
 
     # Save image if it isn't already saved
-    if not os.path.exists(os.path.join("images/", scale.name)):
-        with open(os.path.join("images/", scale.name), "wb") as f:
+    if not os.path.exists(scale.name):
+        with open(scale.name, "wb") as f:
             f.write(scale.getbuffer())
-    scale_pil = Image.open(os.path.join("images/", scale.name))
+    scale_pil = Image.open(scale.name)
 
     # Remove file when done
     os.remove(os.path.join("images/", scale.name))
@@ -123,10 +117,10 @@ if image:
     image_np = cv2.imdecode(image_np, 1)
 
     # Save image if it isn't already saved
-    if not os.path.exists(os.path.join("images/", image.name)):
-        with open(os.path.join("images/", image.name), "wb") as f:
+    if not os.path.exists(image.name):
+        with open(image.name, "wb") as f:
             f.write(image.getbuffer())
-    image_pil = Image.open(os.path.join("images/", image.name))
+    image_pil = Image.open(image.name)
 
     # Remove file when done
     os.remove(os.path.join("images/", image.name))