Update app.py

app.py

@@ -63,7 +63,7 @@ def image_to_text(image, prompt, resolution=1024, steps=64, cfg=9.0):
             width=resolution,
             guidance_scale=cfg,
             num_inference_steps=steps,
-            mask_token_embedding="MeissonFlow/Muddit",
+            mask_token_embedding="MeissonFlow/Muddit/1024",
             generator=torch.manual_seed(42),
         )
 
@@ -84,7 +84,7 @@ def text_to_image(prompt, negative_prompt, num_images=1, resolution=1024, steps=
             width=resolution,
             guidance_scale=cfg,
             num_inference_steps=steps,
-            mask_token_embedding="MeissonFlow/Muddit",
+            mask_token_embedding="MeissonFlow/Muddit/1024",
             generator=torch.manual_seed(42),
         )