handler

handler.py

@@ -7,18 +7,7 @@ class EndpointHandler():
         self.pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
         self.pipeline.load_lora_weights(".", weight_name="./safetensors.safetensors")
 
-        # self.pipeline = StableDiffusionPipeline.from_single_file(
-        #     "https://huggingface.co/artificialguybr/LogoRedmond-LogoLoraForSDXL/blob/main/LogoRedmond_LogoRedAF.safetensors"
-        # )
-
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
-        """
-        data args:
-            inputs (:obj: `str` | `PIL.Image` | `np.array`)
-            kwargs
-        Return:
-            A :obj:`list` | `dict`: will be serialized and returned
-        """
-        # Use self.pipeline for inference
-        # result = self.pipeline(your_input_data)
-        # return result
+        print('data:', data)
+        result = self.pipeline(data)
+        return result