ai-tube-model-als-10

Running on A10G

wangfuyun commited on Feb 27

Commit

e0019a7

•

1 Parent(s): 3c9da29

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ import os
 from glob import glob
 from pathlib import Path
 from typing import Optional
-import time
 from PIL import Image
 from diffusers.utils import load_image, export_to_video
@@ -86,7 +85,7 @@ def sample(
     os.makedirs(output_folder, exist_ok=True)
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
-    video_path = os.path.join(output_folder, f"{base_count:06d}-{int(time.time())}.mp4")
     with torch.autocast("cuda"):
         frames = pipe(

 from glob import glob
 from pathlib import Path
 from typing import Optional
 from PIL import Image
 from diffusers.utils import load_image, export_to_video
     os.makedirs(output_folder, exist_ok=True)
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
+    video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
     with torch.autocast("cuda"):
         frames = pipe(