hlky commited on
Commit
5c16fb4
·
verified ·
1 Parent(s): 5453b53

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -1,6 +1,6 @@
1
  from typing import Dict, List, Any
2
  import torch
3
- from base64 import b64decode
4
  from diffusers import AutoencoderKLHunyuanVideo
5
  from diffusers.video_processor import VideoProcessor
6
  from diffusers.utils import export_to_video
@@ -61,4 +61,4 @@ class EndpointHandler:
61
  with open(path, "rb") as f:
62
  video = f.read()
63
 
64
- return video
 
1
  from typing import Dict, List, Any
2
  import torch
3
+ from base64 import b64decode, b64encode
4
  from diffusers import AutoencoderKLHunyuanVideo
5
  from diffusers.video_processor import VideoProcessor
6
  from diffusers.utils import export_to_video
 
61
  with open(path, "rb") as f:
62
  video = f.read()
63
 
64
+ return {'bytes': b64encode(video).decode("utf-8")}