Update app.py

app.py

@@ -53,13 +53,14 @@ def get_openpos_image(image):
 
 
 def get_depth_image(image):
-    print("Image type", type(image))
+    pil_image = PIL.Image.fromarray(image)
+    print("Image type", type(pil_image))
     depth_estimator = pipeline('depth-estimation')
-    image = depth_estimator(image)['depth']
-    image = np.array(image)
-    image = image[:, :, None]
-    image = np.concatenate([image, image, image], axis=2)
-    return PIL.Image.fromarray(image)
+    pil_image = depth_estimator(pil_image)['depth']
+    pil_image = np.array(pil_image)
+    pil_image = image[:, :, None]
+    pil_image = np.concatenate([pil_image, pil_image, pil_image], axis=2)
+    return PIL.Image.fromarray(pil_image)
 
 
 control_net_modes = {