Fix colorize pipeline call and fallback storage directories

app/colorize_model.py

@@ -230,9 +230,11 @@ class ColorizeModel:
             else:
                 # Use pipeline directly
                 result = self.pipe(
-                    control_image,
                     prompt=prompt,
-                    num_inference_steps=num_inference_steps
+                    image=control_image,
+                    negative_prompt=negative_prompt,
+                    num_inference_steps=num_inference_steps,
+                    guidance_scale=guidance_scale
                 )
                 
                 if isinstance(result, dict) and "images" in result:

app/main.py

@@ -63,10 +63,30 @@ else:
 
 # Always use /data in Spaces (writable). Allow override via env.
 DATA_ROOT = Path(os.getenv("DATA_DIR", "/data"))
-UPLOAD_DIR = Path(os.getenv("UPLOAD_DIR", str(DATA_ROOT / "uploads")))
-RESULT_DIR = Path(os.getenv("RESULT_DIR", str(DATA_ROOT / "results")))
-UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
-RESULT_DIR.mkdir(parents=True, exist_ok=True)
+
+
+def _ensure_dir(preferred: Path, fallback: Path) -> Path:
+    try:
+        preferred.mkdir(parents=True, exist_ok=True)
+        return preferred
+    except Exception as exc:
+        logger.warning("Failed to create directory %s: %s. Falling back to %s", preferred, exc, fallback)
+        try:
+            fallback.mkdir(parents=True, exist_ok=True)
+            return fallback
+        except Exception as fallback_exc:
+            logger.error("Could not create fallback directory %s: %s", fallback, fallback_exc)
+            raise
+
+
+UPLOAD_DIR = _ensure_dir(
+    Path(os.getenv("UPLOAD_DIR", str(DATA_ROOT / "uploads"))),
+    Path("/tmp/colorize_uploads")
+)
+RESULT_DIR = _ensure_dir(
+    Path(os.getenv("RESULT_DIR", str(DATA_ROOT / "results"))),
+    Path("/tmp/colorize_results")
+)
 logger.info("Storage directories -> uploads: %s, results: %s", str(UPLOAD_DIR), str(RESULT_DIR))
 
 # Mount static files for serving results from resolved directories