Fix HF Space build: Remove Predict git dependency and handle gracefully

app.py

@@ -357,8 +357,11 @@ def run_pipeline_action(input_video_path):
 def run_predict_action(pipeline_payload, input_video_path, selection):
     if not pipeline_payload:
         raise gr.Error("Run BADAS + Reason before Predict.")
-    predict_payload, merged_payload = run_predict_only(pipeline_payload, selection=selection, predict_model_name=PREDICT_MODEL_NAME)
-    return build_outputs(merged_payload, "Predict completed.", input_video_path, predict_payload)
+    try:
+        predict_payload, merged_payload = run_predict_only(pipeline_payload, selection=selection, predict_model_name=PREDICT_MODEL_NAME)
+        return build_outputs(merged_payload, "Predict completed.", input_video_path, predict_payload)
+    except Exception as e:
+        raise gr.Error(f"Predict failed: {e}")
 
 
 def safe_warmup_message():

predict_backend.py

@@ -193,8 +193,11 @@ def build_cache_key(source_video_path, badas_context, reason_context, mode, mode
 
 @lru_cache(maxsize=2)
 def get_predict_inference(model_name, output_root_str, disable_guardrails=True):
-    from cosmos_predict2.config import SetupArguments
-    from cosmos_predict2.inference import Inference
+    try:
+        from cosmos_predict2.config import SetupArguments
+        from cosmos_predict2.inference import Inference
+    except ImportError:
+        raise RuntimeError("Cosmos Predict is not installed in this environment.")
 
     setup_args = SetupArguments(
         output_dir=Path(output_root_str),
@@ -241,7 +244,10 @@ def prepare_conditioning_input(source_video_path, badas_context, reason_context,
 
 
 def execute_predict_generation(output_root, model_name, sample_name, conditioning_path, prompt):
-    from cosmos_predict2.config import InferenceArguments
+    try:
+        from cosmos_predict2.config import InferenceArguments
+    except ImportError:
+        raise RuntimeError("Cosmos Predict is not installed in this environment.")
 
     inference = get_predict_inference(model_name, str(output_root), True)
     inference_args = InferenceArguments(

requirements.txt

@@ -13,4 +13,3 @@ accelerate
 torch
 torchvision
 albumentations
-git+https://github.com/nvidia-cosmos/cosmos-predict2.5.git

space_backend.py

@@ -90,7 +90,11 @@ def preload_runtime(preload_badas=True, preload_reason=True, preload_predict=Fal
         steps.append(f"Reason model ready: {reason_model_name}")
     if preload_predict:
         get_predict_inference(predict_model_name, str(PREDICT_OUTPUT_ROOT), True)
-        steps.append(f"Predict model ready: {predict_model_name}")
+        try:
+            get_predict_inference(predict_model_name, str(PREDICT_OUTPUT_ROOT), True)
+            steps.append(f"Predict model ready: {predict_model_name}")
+        except Exception as e:
+            steps.append(f"Predict model skipped: {e}")
     return "\n".join(steps)