Update handler.py

handler.py

@@ -14,11 +14,9 @@ class EndpointHandler:
             use_auth_token=True  # Required for gated base model
         )
 
-        # ✅ Load LoRA from Hugging Face Hub (or local if path is set)
         print("Loading LoRA weights from: Texttra/Cityscape_Studio")
         self.pipe.load_lora_weights("Texttra/Cityscape_Studio", weight_name="c1t3_v1.safetensors")
 
-        # Move to GPU if available
         if torch.cuda.is_available():
             self.pipe.to("cuda")
         else:
@@ -40,21 +38,4 @@ class EndpointHandler:
         print("Extracted prompt:", prompt)
 
         if not prompt:
-            return {"error": "No prompt provided."}
-
-        # ✅ FLUX requires both prompt_embeds and pooled_prompt_embeds
-        conditioning, pooled = self.compel(prompt, return_pooled=True)
-        print("Conditioning complete.")
-
-        image = self.pipe(
-            prompt_embeds=conditioning,
-            pooled_prompt_embeds=pooled
-        ).images[0]
-        print("Image generated.")
-
-        buffer = BytesIO()
-        image.save(buffer, format="PNG")
-        base64_image = base64.b64encode(buffer.getvalue()).decode("utf-8")
-        print("Returning image.")
-
-        return {"image": base64_image}
+            return {"error": "No prompt provided