Update app.py

app.py

@@ -18,7 +18,7 @@ model_arch = "zoe"
 # Inpainting pipeline
 
 
-unet = UNet2DConditionModel.from_pretrained("pablodawson/ldm3d-inpainting", cache_dir="cache", subfolder="unet", in_channels=9, low_cpu_mem_usage=False, ignore_mismatched_sizes=True)
+unet = UNet2DConditionModel.from_pretrained("pablodawson/ldm3d-inpainting", cache_dir="cache", subfolder="unet")
 pipe = StableDiffusionLDM3DInpaintPipeline.from_pretrained("Intel/ldm3d-4c", cache_dir="cache" ).to(device)
 
 # Depth estimation
@@ -74,7 +74,7 @@ elif model_arch == "zoe":
         
         return Image.fromarray(output_.astype("int32")), output.min(), output.max()
 
-def denormalize(image, max, min):
+def denormalize(image, min, max):
     image =  (image / 65535 - 1 ) * (min - max) + min
     return image