update demo

app.py

@@ -53,23 +53,6 @@ from datetime import datetime, timedelta
 import spaces
 import moviepy.editor as mp
 
-
-import os
-from huggingface_hub import login
-login(token=os.getenv('HF_TOKEN'))
-
-dtype = torch.float16
-device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = VchitectXLPipeline("Vchitect/Vchitect-XL-2B",device)
-
-# pipe.acc_call = acc_call.__get__(pipe)
-import types
-# pipe.__call__ = types.MethodType(acc_call, pipe)
-pipe.__class__.__call__ = acc_call
-
-os.makedirs("./output", exist_ok=True)
-os.makedirs("./gradio_tmp", exist_ok=True)
-
 # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.retrieve_timesteps
 def retrieve_timesteps(
     scheduler,
@@ -357,6 +340,23 @@ def acc_call(
             videos.append(image[0])
         
         return videos
+        
+import os
+from huggingface_hub import login
+login(token=os.getenv('HF_TOKEN'))
+
+dtype = torch.float16
+device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = VchitectXLPipeline("Vchitect/Vchitect-XL-2B",device)
+
+# pipe.acc_call = acc_call.__get__(pipe)
+import types
+# pipe.__call__ = types.MethodType(acc_call, pipe)
+pipe.__class__.__call__ = acc_call
+
+os.makedirs("./output", exist_ok=True)
+os.makedirs("./gradio_tmp", exist_ok=True)
+
 
 @spaces.GPU(duration=120)
 def infer(prompt: str, progress=gr.Progress(track_tqdm=True)):