int32 depth

app.py

@@ -51,9 +51,9 @@ def estimate_depth(image):
 
     output = prediction.cpu().numpy()
 
-    output= 255 * (output - np.min(output))/(np.max(output) - np.min(output))
+    output= 65535 * (output - np.min(output))/(np.max(output) - np.min(output))
     
-    return Image.fromarray(output.astype("uint8"))
+    return Image.fromarray(output.astype("int32"))
 
 
 def read_content(file_path: str) -> str:
@@ -78,8 +78,10 @@ def predict(dict, depth, prompt="", negative_prompt="", guidance_scale=7.5, step
         depth_image = estimate_depth(init_image)
         
     else:
-        depth_image = depth
-        depth_image = Image.fromarray(depth_image[:,:,0].astype("uint8"))
+        d_i = depth[:,:,0]
+        depth_image = 65535 * (d_i - np.min(d_i))/(np.max(d_i) - np.min(d_i))
+        depth_image = depth_image.astype("int32")
+        depth_image = Image.fromarray(depth_image)
 
     init_image = Image.fromarray(init_image.astype("uint8"))
     init_image.save("temp_image.jpg")