Update app.py

app.py

@@ -49,4 +49,93 @@ def download_model():
         print(f"❌ FP8 Checkpoint missing at {CHECKPOINT_FP8_FILE}")
         sys.exit(1)
 
-    print("✅ Model downloaded suc
+    print("✅ Model downloaded successfully.")
+    clear_hf_cache()
+
+def run_sample_gpu_poor():
+    print("🎬 Running sample_gpu_poor.py...")
+
+    cmd = [
+        "python3", "hymm_sp/sample_gpu_poor.py",
+        "--input", "assets/test.csv",
+        "--ckpt", CHECKPOINT_FP8_FILE,
+        "--sample-n-frames", "129",
+        "--seed", "128",
+        "--image-size", "704",
+        "--cfg-scale", "7.5",
+        "--infer-steps", "50",
+        "--use-deepcache", "1",
+        "--flow-shift-eval-video", "5.0",
+        "--save-path", OUTPUT_BASEPATH,
+        "--use-fp8",
+        "--cpu-offload",
+        "--infer-min"
+    ]
+
+    env = os.environ.copy()
+    env.update(HF_HOME_ENV)
+    env["PYTHONPATH"] = "./"
+    env["MODEL_BASE"] = WEIGHTS_DIR
+    env["CPU_OFFLOAD"] = "1"
+    env["CUDA_VISIBLE_DEVICES"] = "0"
+
+    result = subprocess.run(cmd, env=env)
+    if result.returncode != 0:
+        print("❌ sample_gpu_poor.py failed.")
+        sys.exit(1)
+
+    print("✅ sample_gpu_poor.py completed.")
+
+def run_flask_audio():
+    print("🚀 Starting flask_audio.py...")
+    cmd = [
+        "torchrun",
+        "--nnodes=1",
+        "--nproc_per_node=8",
+        "--master_port=29605",
+        "hymm_gradio/flask_audio.py",
+        "--input", "assets/test.csv",
+        "--ckpt", CHECKPOINT_FILE,
+        "--sample-n-frames", "129",
+        "--seed", "128",
+        "--image-size", "704",
+        "--cfg-scale", "7.5",
+        "--infer-steps", "50",
+        "--use-deepcache", "1",
+        "--flow-shift-eval-video", "5.0"
+    ]
+    subprocess.Popen(cmd)
+
+def run_gradio_ui():
+    print("🟢 Launching Gradio UI...")
+    cmd = ["python3", "hymm_gradio/gradio_audio.py"]
+    subprocess.Popen(cmd)
+
+def create_gitignore():
+    gitignore_path = os.path.join(BASE_DIR, ".gitignore")
+    lines_to_add = ["weights/", "results-*/", ".cache/"]
+    with open(gitignore_path, "a+") as f:
+        f.seek(0)
+        existing = f.read().splitlines()
+        for line in lines_to_add:
+            if line not in existing:
+                f.write(f"{line}\n")
+
+def main():
+    create_gitignore()
+    check_disk_space(min_free_gb=10)
+
+    if os.path.isfile(CHECKPOINT_FILE) and os.path.isfile(CHECKPOINT_FP8_FILE):
+        print("✅ Checkpoints exist. Skipping download.")
+    else:
+        download_model()
+
+    run_sample_gpu_poor()
+
+    # Optional: Launch UIs
+    run_flask_audio()
+    time.sleep(5)
+    run_gradio_ui()
+
+if __name__ == "__main__":
+    main()