change storage place

app.py

@@ -12,7 +12,7 @@ def calculate_depth(model_type, img):
     if not os.path.exists('temp'):
       os.system('mkdir temp')
 
-    filename = "temp/image.jpg"
+    filename = "Images/Input-Test/0.jpg"
 
     img.save(filename, "JPEG")
 
@@ -46,11 +46,12 @@ def calculate_depth(model_type, img):
 
     output = prediction.cpu().numpy()
 
-    formatted = (output * 255 / np.max(output)).astype('uint8')
-    out_im = Image.fromarray(formatted)
-    out_im.save("temp/image_depth.jpeg", "JPEG")
+    out_im = Image.fromarray(np.uint8(output*255))
+    #formatted = (output * 255 / np.max(output)).astype('uint8')
+    #out_im = Image.fromarray(formatted)
+    out_im.save("Images/Input-Test/0_d.jpg", "JPEG")
 
-    return f'temp/image_depth.jpeg'
+    return f'Images/Input-Test/0_d.jpg'
 
 def wiggle_effect(slider):