Add application file

app.py

@@ -58,7 +58,7 @@ def load_model(model_name):
     return current_model.to('cuda')
 
 
-@spaces.GPU(duration=1000)
+@spaces.GPU(duration=500)
 def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50, use_cfg_zero_star=True, use_zero_init=True, zero_steps=0, seed=None, compare_mode=False):
     model = load_model(model_name)
     if seed is None:
@@ -81,19 +81,24 @@ def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps
             video1_path = os.path.join(OUTPUT_DIR, f"{seed}_CFG-Zero-Star.mp4")
             export_to_video(video1_frames, video1_path, fps=16)
 
-            set_seed(seed)
-            video2_frames = model(
-                prompt=prompt,
-                guidance_scale=guidance_scale,
-                num_frames=81,
-                use_cfg_zero_star=False,
-                use_zero_init=use_zero_init,
-                zero_steps=zero_steps
-            ).frames[0]
-            video2_path = os.path.join(OUTPUT_DIR,  f"{seed}_CFG.mp4")
-            export_to_video(video2_frames, video2_path, fps=16)
-
-            return None, None, video1_path, video2_path, seed
+            if use_cfg_zero_star:
+                return None, None, video1_path, None, seed
+            else:
+                return None, None, None, video1_path, seed
+        
+            # set_seed(seed)
+            # video2_frames = model(
+            #     prompt=prompt,
+            #     guidance_scale=guidance_scale,
+            #     num_frames=81,
+            #     use_cfg_zero_star=False,
+            #     use_zero_init=use_zero_init,
+            #     zero_steps=zero_steps
+            # ).frames[0]
+            # video2_path = os.path.join(OUTPUT_DIR,  f"{seed}_CFG.mp4")
+            # export_to_video(video2_frames, video2_path, fps=16)
+
+            # return None, None, video1_path, video2_path, seed
         # else:
         #     video_frames = model(
         #         prompt=prompt,