Update app.py

app.py

@@ -398,7 +398,7 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     pipe = YOSODiffusePipeline.from_pretrained(
-        'Stable-X/yoso-delight-v0-4', trust_remote_code=True, variant="fp16", 
+        'Stable-X/yoso-delight-v0-4-base', trust_remote_code=True, variant="fp16", 
         torch_dtype=torch.float16, t_start=0).to(device)
     try:
         import xformers