Update handler.py

handler.py

@@ -10,6 +10,8 @@ import numpy as np
 from moviepy.editor import ImageSequenceClip
 import os
 import logging
+import asyncio
+from varnish import Varnish
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -24,6 +26,11 @@ generator = torch.manual_seed(0)
 # you can notice we don't use device=cuda, for more info see:
 # https://huggingface.co/docs/diffusers/v0.16.0/en/using-diffusers/reproducibility#gpu
 
+varnish = Varnish(
+    enable_mmaudio=False,
+    #mmaudio_config=mmaudio_config
+)
+
 class EndpointHandler:
     # Default configuration
     DEFAULT_FPS = 24
@@ -135,7 +142,7 @@ class EndpointHandler:
         
         # Create temporary file
         output_path = tempfile.mktemp(suffix=".mp4")
-        
+
         try:
             # Create video clip and write to file
             clip = ImageSequenceClip(list(video_np), fps=fps)