Update app.py

app.py

@@ -7,7 +7,12 @@ import torch
 from diffusers import Flux2KleinPipeline
 from huggingface_hub import login, hf_hub_download
 from safetensors.torch import load_file
-from color_matcher import ColorMatcher
+try:
+    from color_matcher import ColorMatcher
+except ImportError:
+    import subprocess
+    subprocess.check_call(["pip", "install", "color-matcher"])
+    from color_matcher import ColorMatcher
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"