Update handler.py

handler.py

@@ -4,7 +4,6 @@ from diffusers import DiffusionPipeline
 from compel import Compel
 from io import BytesIO
 import base64
-import os
 
 class EndpointHandler:
     def __init__(self, path: str = ""):
@@ -12,12 +11,12 @@ class EndpointHandler:
         self.pipe = DiffusionPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.float16,
-            use_auth_token=True
+            use_auth_token=True  # Safe since base model is gated
         )
 
-        lora_path = os.path.join(path, "c1t3_v1.safetensors")
-        print(f"Loading LoRA weights from: {lora_path}")
-        self.pipe.load_lora_weights(lora_path)
+        # ✅ Load LoRA directly from Hugging Face Hub
+        print("Loading LoRA weights from: Texttra/Cityscape_Studio")
+        self.pipe.load_lora_weights("Texttra/Cityscape_Studio", weight_name="c1t3_v1.safetensors")
 
         if torch.cuda.is_available():
             self.pipe.to("cuda")
@@ -53,4 +52,4 @@ class EndpointHandler:
         base64_image = base64.b64encode(buffer.getvalue()).decode("utf-8")
         print("Returning image.")
 
-        return {"image": base64_image}
+        return {"image": base64_image}