Update app.py

app.py

@@ -15,7 +15,7 @@ from diffusers import HunyuanVideoTransformer3DModel
 from diffusers.utils import export_to_video
 from diffusers.utils import load_image
 from PIL import Image
-
+import numpy as np
 from torchao.quantization import float8_weight_only
 from torchao.quantization import quantize_
 from transformers import LlamaModel
@@ -80,13 +80,14 @@ negative_prompt = "Aerial view, aerial view, overexposed, low quality, deformati
 
 @spaces.GPU(duration=60)
 def generate(segment, image, prompt, size, guidance_scale, num_inference_steps, frames, seed, progress=gr.Progress(track_tqdm=True) ):
-    Offload.offload(
-        pipeline=pipe,
-        config=offload_config,
-    )
+
     random.seed(time.time())
     seed = int(random.randrange(4294967294))
     if segment==1:
+        Offload.offload(
+            pipeline=pipe,
+            config=offload_config,
+        )
         prompt_embeds, prompt_attention_mask, negative_prompt_embeds, negative_attention_mask, pooled_prompt_embeds, negative_pooled_prompt_embeds = pipe.encode_prompt(
         prompt=prompt, do_classifier_free_guidance=True, negative_prompt=negative_prompt, device=device
         )