Update app.py

app.py

@@ -14,7 +14,6 @@ import numpy as np
 from PIL import Image
 import random
 import gc
-from optimization import optimize_pipeline_
 
 
 # =========================================================
@@ -41,12 +40,14 @@ MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS, 1)
 # =========================================================
 # MODEL LOADING
 # =========================================================
+print("Loading VAE...")
 vae = AutoencoderKLWan.from_pretrained(
     "Wan-AI/Wan2.2-T2V-A14B-Diffusers", 
     subfolder="vae", 
     torch_dtype=torch.float32
 )
 
+print("Loading pipeline...")
 pipe = WanPipeline.from_pretrained(
     MODEL_ID,
     transformer=WanTransformer3DModel.from_pretrained(
@@ -65,18 +66,13 @@ pipe = WanPipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
 ).to('cuda')
 
+# Memory cleanup
 for i in range(3): 
     gc.collect()
     torch.cuda.synchronize() 
     torch.cuda.empty_cache()
 
-optimize_pipeline_(
-    pipe,
-    prompt='prompt',
-    height=LANDSCAPE_HEIGHT,
-    width=LANDSCAPE_WIDTH,
-    num_frames=MAX_FRAMES_MODEL,
-)
+print("Pipeline loaded successfully!")
 
 # =========================================================
 # DEFAULT PROMPTS
@@ -96,10 +92,6 @@ Rules:
 4. Keep the enhanced prompt concise but detailed (max 150 words)
 5. Maintain the original intent of the user's prompt
 6. Output ONLY the enhanced prompt, nothing else
-
-Example:
-User: "A cat playing"
-Enhanced: "A fluffy orange tabby cat playfully batting at a dangling yarn ball, soft afternoon sunlight streaming through a window creating warm golden highlights on its fur, smooth tracking shot following the cat's graceful movements, shallow depth of field with bokeh background, cozy living room setting with warm ambient lighting"
 """
 
 
@@ -180,8 +172,8 @@ def generate_video(
     output_frames_list = pipe(
         prompt=final_prompt,
         negative_prompt=negative_prompt,
-        height=480,
-        width=832,
+        height=LANDSCAPE_HEIGHT,
+        width=LANDSCAPE_WIDTH,
         num_frames=num_frames,
         guidance_scale=float(guidance_scale),
         guidance_scale_2=float(guidance_scale_2),
@@ -202,7 +194,7 @@ def generate_video(
    • Duration: {actual_duration:.2f} seconds
    • Total Frames: {num_frames}
    • FPS: {FIXED_FPS}
-   • Resolution: 832 x 480
+   • Resolution: {LANDSCAPE_WIDTH} x {LANDSCAPE_HEIGHT}
 {'=' * 50}
 ⚙️ Generation Settings:
    • Guidance Scale: {guidance_scale} / {guidance_scale_2}