Upload folder using huggingface_hub

handler.py

@@ -56,7 +56,6 @@ class EndpointHandler:
         """
         inputs = data.pop("inputs", data)
         request = ImageRequest.FromDict(inputs)
-        self.LoadModel(request)
         response = self.__runProcess__(request)
         return response
     
@@ -75,9 +74,10 @@ class EndpointHandler:
         import torch
         
 
+        self.LoadModel(request)
         
 
         # Ensure using the same inference steps as the loaded model and CFG set to 0.
-        images = pipe(request.prompt, negative_prompt = request.negative_prompt, num_inference_steps=request.steps, guidance_scale=0).images
+        images = self.pipe(request.prompt, negative_prompt = request.negative_prompt, num_inference_steps=request.steps, guidance_scale=0).images
 
         return {"media":[self.ImageToBase64(img) for img in images]}