Update app.py

app.py

@@ -40,19 +40,19 @@ ORIGINAL_MAX_DURATION = round(81/24, 1)  # MAX_FRAMES_MODEL/FIXED_FPS
 # Limited space constants
 LIMITED_MAX_RESOLUTION = 640
 LIMITED_MAX_DURATION = 2.0
-LIMITED_MAX_STEPS = 3
+LIMITED_MAX_STEPS = 4
 
 # Set limits based on environment variable
 if IS_ORIGINAL_SPACE:
-    SLIDER_MIN_H, SLIDER_MAX_H = ORIGINAL_SLIDER_MIN_H, ORIGINAL_SLIDER_MAX_H
-    SLIDER_MIN_W, SLIDER_MAX_W = ORIGINAL_SLIDER_MIN_W, ORIGINAL_SLIDER_MAX_W
-    MAX_DURATION = ORIGINAL_MAX_DURATION
-    MAX_STEPS = 8
-else:
     SLIDER_MIN_H, SLIDER_MAX_H = 128, LIMITED_MAX_RESOLUTION
     SLIDER_MIN_W, SLIDER_MAX_W = 128, LIMITED_MAX_RESOLUTION
     MAX_DURATION = LIMITED_MAX_DURATION
     MAX_STEPS = LIMITED_MAX_STEPS
+else:
+    SLIDER_MIN_H, SLIDER_MAX_H = ORIGINAL_SLIDER_MIN_H, ORIGINAL_SLIDER_MAX_H
+    SLIDER_MIN_W, SLIDER_MAX_W = ORIGINAL_SLIDER_MIN_W, ORIGINAL_SLIDER_MAX_W
+    MAX_DURATION = ORIGINAL_MAX_DURATION
+    MAX_STEPS = 8
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -125,7 +125,7 @@ def generate_video(prompt, height, width,
         raise gr.Error("Please enter a text prompt. Try to use long and precise descriptions.")
 
     # Apply limits based on environment variable
-    if not IS_ORIGINAL_SPACE:
+    if IS_ORIGINAL_SPACE:
         height = min(height, LIMITED_MAX_RESOLUTION)
         width = min(width, LIMITED_MAX_RESOLUTION)
         duration_seconds = min(duration_seconds, LIMITED_MAX_DURATION)
@@ -156,7 +156,7 @@ with gr.Blocks() as demo:
     gr.Markdown("This Gradio space is a fork of [wan2-1-fast from multimodalart](https://huggingface.co/spaces/multimodalart/wan2-1-fast), and is powered by the Wan CausVid LoRA [from Kijai](https://huggingface.co/Kijai/WanVideo_comfy/blob/main/Wan21_CausVid_bidirect2_T2V_1_3B_lora_rank32.safetensors).")
 
     # Add notice for limited spaces
-    if not IS_ORIGINAL_SPACE:
+    if IS_ORIGINAL_SPACE:
         gr.Markdown("⚠️ **This free public demo limits the resolution to 640x640, duration to 2s, and inference steps to 3. For full capabilities please duplicate this space.**")
     
     with gr.Row():
@@ -211,7 +211,7 @@ with gr.Blocks() as demo:
         ["a field of flowers swaying in the wind, spring morning light", 512, 896],
     ]
     
-    if not IS_ORIGINAL_SPACE:
+    if IS_ORIGINAL_SPACE:
         # Limit example resolutions for limited spaces
         example_configs = [
             [example[0], min(example[1], LIMITED_MAX_RESOLUTION), min(example[2], LIMITED_MAX_RESOLUTION)]