Save generated video to /data/output

start.sh

@@ -1,16 +1,21 @@
 #!/bin/bash
 
 # Set environment variables for a single GPU on Hugging Face Spaces
-export CUDA_VISIBLE_DEVICES="0" 
-export CUDA_HOME="/usr/local/cuda" 
+export CUDA_VISIBLE_DEVICES="0"
+export CUDA_HOME="/usr/local/cuda"
 export PYTHONPATH="/app"
 
 echo "Starting GEN3C application on A100 Large 80GB GPU..."
 
+# Create the /data directory if it doesn't exist (it should be mounted, but good practice)
+mkdir -p /data/output
+
+# Run your Python script, saving the output video to /data/output/
+# IMPORTANT: Ensure your script can handle an absolute path for --video_save_name
 python cosmos_predict1/diffusion/inference/gen3c_single_image.py \
   --checkpoint_dir checkpoints \
   --input_image_path assets/diffusion/000000.png \
-  --video_save_name test_single_image \
+  --video_save_name /data/output/test_single_image.mp4 \ # <-- CHANGED PATH HERE
   --guidance 1 \
   --foreground_masking \
   --offload_diffusion_transformer \
@@ -21,7 +26,5 @@ python cosmos_predict1/diffusion/inference/gen3c_single_image.py \
   --disable_guardrail \
   --disable_prompt_encoder
 
-# IMPORTANT: If your Python script (gen3c_single_image.py) is designed to run and then exit,
-# your Hugging Face Space will stop after it finishes.
-# If your application is meant to be a continuous service (like a Gradio/Streamlit app),
-# ensure that the Python script itself keeps running (e.g., by starting a web server).
+# IMPORTANT: Your Space will still stop after this script finishes,
+# but the generated video will now be saved in /data/output/ and persist.