Update config.py

config.py

@@ -2,12 +2,16 @@ import torch
 
 class Config:
     # Hardware
-    DEVICE = "cuda" if torch.cuda.is.available() else "cpu"
+    if torch.cuda.is_available():
+        DEVICE = "cuda"
+    else:
+        DEVICE = "cpu"
+        
     DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 
     # Base Model & LoRA (from primerz/pixagram)
     REPO_ID = "primerz/pixagram"
-    CHECKPOINT_FILENAME = "horizon.safensors"
+    CHECKPOINT_FILENAME = "horizon.safetensors"
     LORA_FILENAME = "retroart.safetensors"
     LORA_STRENGTH = 1.2
     
@@ -32,7 +36,7 @@ class Config:
     
     # InsightFace Model (HF Hub mirror)
     ANTELOPEV2_REPO = "DIAMONIK7777/antelopev2"
-    ANTELOPEV2_ROOT = "." # Parent folder
+    ANTELOPEV2_ROOT = "."
     ANTELOPEV2_NAME = "antelopev2"
     
     # Gradio Parameters