ldm3d-inpainting

Runtime error

pablo commited on Sep 30, 2023

Commit

aae12bf

1 Parent(s): 8603848

???

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ else:
     transform = midas_transforms.small_transform
-def estimate_depth(image: Image) -> Image:
     image= image.resize((384,384)).convert("RGB")
     image = np.array(image)
     input_batch = transform(image).to(device)

     transform = midas_transforms.small_transform
+def estimate_depth(image):
     image= image.resize((384,384)).convert("RGB")
     image = np.array(image)
     input_batch = transform(image).to(device)