fix depth issue

app.py

@@ -46,16 +46,15 @@ def calculate_depth(model_type, img):
 
     output = prediction.cpu().numpy()
 
-    out_im = Image.fromarray(np.uint8(output*255))
-    #formatted = (output * 255 / np.max(output)).astype('uint8')
-    #out_im = Image.fromarray(formatted)
+    formatted = (output * 255.0 / np.max(output)).astype('uint8')
+    out_im = Image.fromarray(formatted)
     out_im.save("Images/Input-Test/0_d.jpg", "JPEG")
 
     return f'Images/Input-Test/0_d.jpg'
 
 def wiggle_effect(slider):
 
-    return [f'temp/image_depth.jpeg',f'temp/image_depth.jpeg']
+    return [f'Images/Input-Test/0_d.jpg',f'Images/Input-Test/0_d.jpg']
 
 
 #pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116