Update src/florence/model.py

src/florence/model.py

@@ -1,4 +1,5 @@
 # Importing necessary libraries
+import os
 import sys
 import subprocess
 from typing import Optional
@@ -14,9 +15,15 @@ from src.exception import CustomExceptionHandling
 
 
 # Install the required dependencies
+# NOTE: passing env={...} to subprocess.run REPLACES the entire environment
+# instead of extending it, which strips PATH/HOME/etc. and can break the
+# pip install (and therefore the whole Space) at import time. Copy the
+# current environment first and only add/override the flag we need.
+env = os.environ.copy()
+env["FLASH_ATTENTION_SKIP_CUDA_BUILD"] = "TRUE"
 subprocess.run(
     "pip install flash-attn --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    env=env,
     shell=True,
 )