Update app.py

app.py

@@ -2,8 +2,6 @@
 import os
 os.system('pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces')
 
-assert False
-
 # Actual demo code
 import spaces
 import torch
@@ -25,13 +23,9 @@ LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
 
 image_encoder = CLIPVisionModel.from_pretrained(MODEL_ID, subfolder="image_encoder", torch_dtype=torch.float32)
 vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float32)
-
-with tempfile.TemporaryDirectory() as temporary_directory:
-    pipe = WanImageToVideoPipeline.from_pretrained(
-        MODEL_ID, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16,
-        cache_dir=temporary_directory,
-    )
-
+pipe = WanImageToVideoPipeline.from_pretrained(
+    MODEL_ID, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16,
+)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=8.0)
 pipe.to("cuda")