lemonaddie commited on
Commit
61a06ab
1 Parent(s): 9d8a47d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ vae = AutoencoderKL.from_pretrained('.', subfolder='vae')
49
  scheduler = DDIMScheduler.from_pretrained('.', subfolder='scheduler')
50
  image_encoder = CLIPVisionModelWithProjection.from_pretrained('.', subfolder="image_encoder")
51
  feature_extractor = CLIPImageProcessor.from_pretrained('.', subfolder="feature_extractor")
52
- unet = UNet2DConditionModel.from_pretrained('.', subfolder="unet")
53
 
54
  pipe = DepthNormalEstimationPipeline(vae=vae,
55
  image_encoder=image_encoder,
 
49
  scheduler = DDIMScheduler.from_pretrained('.', subfolder='scheduler')
50
  image_encoder = CLIPVisionModelWithProjection.from_pretrained('.', subfolder="image_encoder")
51
  feature_extractor = CLIPImageProcessor.from_pretrained('.', subfolder="feature_extractor")
52
+ unet = UNet2DConditionModel.from_pretrained('.', subfolder="unet_7000")
53
 
54
  pipe = DepthNormalEstimationPipeline(vae=vae,
55
  image_encoder=image_encoder,