Add application file

app.py

@@ -68,7 +68,7 @@ def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps
     is_video_model = "wan-t2v" in model_name
 
     if is_video_model:
-        if compare_mode:
+        if True:
             set_seed(seed)
             video1_frames = model(
                 prompt=prompt,
@@ -94,17 +94,17 @@ def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps
             export_to_video(video2_frames, video2_path, fps=16)
 
             return None, None, video1_path, video2_path, seed
-        else:
-            video_frames = model(
-                prompt=prompt,
-                guidance_scale=guidance_scale,
-                num_frames=81,
-                use_cfg_zero_star=use_cfg_zero_star,
-                use_zero_init=use_zero_init,
-                zero_steps=zero_steps
-            ).frames[0]
-            video_path = save_video(video_frames, f"{seed}.mp4")
-            return None, None, video_path, None, seed
+        # else:
+        #     video_frames = model(
+        #         prompt=prompt,
+        #         guidance_scale=guidance_scale,
+        #         num_frames=81,
+        #         use_cfg_zero_star=use_cfg_zero_star,
+        #         use_zero_init=use_zero_init,
+        #         zero_steps=zero_steps
+        #     ).frames[0]
+        #     video_path = save_video(video_frames, f"{seed}.mp4")
+        #     return None, None, video_path, None, seed
 
     if compare_mode:
         set_seed(seed)
@@ -146,7 +146,7 @@ def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps
 demo = gr.Interface(
     fn=generate_content,
     inputs=[
-        gr.Textbox(value="A cosmic whale swimming throught a glaxy with stars and swirling cosmic dusts.", label="Enter your prompt"),
+        gr.Textbox(value="A spooky haunted mansion on a hill silhouetted by a full moon.", label="Enter your prompt"),
         gr.Dropdown(choices=list(model_paths.keys()), label="Choose Model"),
         gr.Slider(1, 20, value=4.0, step=0.5, label="Guidance Scale"),
         gr.Slider(10, 100, value=28, step=5, label="Inference Steps"),