Update app.py

app.py

@@ -13,7 +13,7 @@ import uuid
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
-is_canonical = True if os.environ.get("SPACE_ID") == "multimodalart/pyramid-flow" else False
+is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow" else False
 
 # Constants
 MODEL_PATH = "pyramid-flow-model"
@@ -122,7 +122,7 @@ with gr.Blocks() as demo:
             gr.HTML("""
                 <div style="display: flex; flex-direction: column;justify-content: center; align-items: center; text-align: center;">
                     <p style="display: flex;gap: 6px;">
-                         <a href="https://huggingface.co/spaces/multimodalart/pyramid-flow?duplicate=true">
+                         <a href="https://huggingface.co/spaces/Pyramid-Flow/pyramid-flow?duplicate=true">
                             <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-lg.svg" alt="Duplicate this Space">
                         </a>
                     </p>