Upload folder using huggingface_hub

handler.py

@@ -79,4 +79,4 @@ class EndpointHandler:
         # Ensure using the same inference steps as the loaded model and CFG set to 0.
         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]}
+        return {"media":[{"media":self.ImageToBase64(img)} for img in images]}