Update config.py

config.py

@@ -15,12 +15,15 @@ HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN", None)  # Get token from envir
 # Model files
 MODEL_FILES = {
     "checkpoint": "horizon.safetensors",
-    "lora": "retroart.safetensors",
+    "lora_retroart": "retroart.safetensors",
+    "lora_vga": "vga.safetensors",
+    "lora_lucasart": "lucasart.safetensors",
     "vae": "pixelate.safetensors"
 }
 
 # LORA configuration
 TRIGGER_WORD = "p1x3l4rt, pixel art"
+LORA_CHOICES = ["RetroArt", "VGA", "LucasArt", "None"]
 
 # Face detection configuration
 FACE_DETECTION_CONFIG = {
@@ -46,6 +49,7 @@ DEFAULT_PARAMS = {
     "depth_control_scale": 0.75,
     "identity_control_scale": 0.85,
     "expression_control_scale": 0.6,  # <-- NEW
+    "lora_choice": "RetroArt",
     "lora_scale": 1.0,
     "identity_preservation": 1.2,
     "enable_color_matching": False,