Update app.py

app.py

@@ -69,7 +69,7 @@ if scale:
     scale_pil = Image.open(scale.name)
 
     # Remove file when done
-    os.remove(os.path.join("images/", scale.name))
+    os.remove(scale.name)
 
     #inputs = processor(raw_image, return_tensors="pt").to(device)
     inputs = processor(scale_np, return_tensors="pt").to(device)
@@ -123,7 +123,7 @@ if image:
     image_pil = Image.open(image.name)
 
     # Remove file when done
-    os.remove(os.path.join("images/", image.name))
+    os.remove(image.name)
 
     #inputs = processor(raw_image, return_tensors="pt").to(device)
     inputs = processor(image_np, return_tensors="pt").to(device)