Upload 4 files

app.py

@@ -0,0 +1,36 @@
+import streamlit as st
+import torch
+import subprocess
+import os
+
+# Check if GPU is available
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Title
+st.title("WAN 2.1 - 1.3B Text-to-Video Generator 🎥")
+
+# Model selection
+model_path = "./Wan2.1-T2V-1.3B"
+
+# Input fields
+prompt = st.text_area("Enter your text prompt:", "A cat in military dress wearing headphones, laughing and walking.")
+frame_num = st.slider("Number of frames:", min_value=30, max_value=180, value=60, step=10)
+resolution = st.selectbox("Select resolution:", ["832*480", "1280*720"])
+sample_steps = st.slider("Sampling steps:", min_value=10, max_value=50, value=30, step=5)
+
+# Button to generate video
+if st.button("Generate Video"):
+    st.info("Generating video... Please wait.")
+
+    # Run WAN 2.1 with user settings
+    command = f"python generate.py --task t2v-1.3B --size {resolution} --frame_num {frame_num} --sample_steps {sample_steps} --ckpt_dir {model_path} --prompt \"{prompt}\""
+    
+    process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    process.wait()
+
+    # Display video if generated
+    if os.path.exists("output.mp4"):
+        st.video("output.mp4")
+        st.success("✅ Video generated successfully!")
+    else:
+        st.error("❌ Video generation failed. Check logs for details.")

generate.py

@@ -0,0 +1,29 @@
+import argparse
+import torch
+import subprocess
+import os
+
+# Define Arguments
+parser = argparse.ArgumentParser()
+parser.add_argument("--task", type=str, default="t2v-1.3B")
+parser.add_argument("--size", type=str, default="832*480")
+parser.add_argument("--frame_num", type=int, default=60)
+parser.add_argument("--sample_steps", type=int, default=30)
+parser.add_argument("--ckpt_dir", type=str, default="./Wan2.1-T2V-1.3B")
+parser.add_argument("--prompt", type=str, required=True)
+args = parser.parse_args()
+
+# Check GPU Availability
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
+# Run WAN 2.1 Inference
+command = f"python run_model.py --task {args.task} --size {args.size} --frame_num {args.frame_num} --sample_steps {args.sample_steps} --ckpt_dir {args.ckpt_dir} --prompt \"{args.prompt}\" --device {device}"
+
+subprocess.run(command, shell=True)
+
+# Save output
+if os.path.exists("output.mp4"):
+    print("✅ Video generated successfully: output.mp4")
+else:
+    print("❌ Error generating video.")

requirements.txt

@@ -0,0 +1,7 @@
+streamlit
+torch
+transformers
+diffusers
+opencv-python
+pillow
+huggingface_hub