Update app.py

app.py

@@ -376,12 +376,12 @@ text_encoder21 = CLIPTextModel.from_pretrained(
 
 vae15 = AutoencoderKL.from_pretrained('runwayml/stable-diffusion-v1-5', subfolder="vae", revision=args.revision).cuda()
 unet15 = UNet2DConditionModel.from_pretrained(
-    'textdiffuser-ckpt/diffusion_backbone_1.5', subfolder="unet", revision=None 
+    './textdiffuser-ckpt/diffusion_backbone_1.5', subfolder="unet", revision=None 
 ).cuda() 
 
 vae21 = AutoencoderKL.from_pretrained('stabilityai/stable-diffusion-2-1', subfolder="vae", revision=args.revision).cuda()
 unet21 = UNet2DConditionModel.from_pretrained(
-    'textdiffuser-ckpt/diffusion_backbone_2.1', subfolder="unet", revision=None 
+    './textdiffuser-ckpt/diffusion_backbone_2.1', subfolder="unet", revision=None 
 ).cuda() 
 
 scheduler15 = DDPMScheduler.from_pretrained('runwayml/stable-diffusion-v1-5', subfolder="scheduler")