Update app.py
Browse files
app.py
CHANGED
|
@@ -49,8 +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 |
-
|
| 53 |
-
unet = UNet2DConditionModel.from_pretrained('./unet')
|
| 54 |
|
| 55 |
pipe = DepthNormalEstimationPipeline(vae=vae,
|
| 56 |
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")
|
|
|
|
| 53 |
|
| 54 |
pipe = DepthNormalEstimationPipeline(vae=vae,
|
| 55 |
image_encoder=image_encoder,
|