Update app.py

app.py

@@ -201,7 +201,7 @@ def load_ckpt():
             ])
     # seed = 43
     lora_rank = 128
-    pretrained_model_name_or_path = '/mnt/workspace/zhuangjunhao/ColorFlow/PixArt-XL-2-1024-MS/models--PixArt-alpha--PixArt-XL-2-1024-MS/snapshots/b89adadeccd9ead2adcb9fa2825d3fabec48d404'
+    pretrained_model_name_or_path = "PixArt-alpha/PixArt-XL-2-1024-MS"
 
     transformer = PixArtTransformer2DModel.from_pretrained(
             pretrained_model_name_or_path, subfolder="transformer", revision=None, variant=None