Update handler.py

handler.py

@@ -14,7 +14,6 @@ from diffusers import LTXPipeline, LTXImageToVideoPipeline
 from PIL import Image
 
 from varnish import Varnish
-from varnish.debug_utils import setup_debug_logging
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -25,6 +24,22 @@ MAX_WIDTH = 1280
 MAX_HEIGHT = 720
 MAX_FRAMES = 257
 
+def print_directory_structure(startpath):
+    """Print the directory structure starting from the given path."""
+    for root, dirs, files in os.walk(startpath):
+        level = root.replace(startpath, '').count(os.sep)
+        indent = ' ' * 4 * level
+        logger.info(f"{indent}{os.path.basename(root)}/")
+        subindent = ' ' * 4 * (level + 1)
+        for f in files:
+            logger.info(f"{subindent}{f}")
+
+logger.info("💡 Printing directory structure of ""/repository"":")
+print_directory_structure("/repository")
+
+logger.info("💡 Printing directory structure of os.getcwd():")
+print_directory_structure(os.getcwd())
+
 @dataclass
 class GenerationConfig:
     """Configuration for video generation"""
@@ -87,9 +102,6 @@ class EndpointHandler:
         #self.text_to_video.enable_model_cpu_offload()
         #self.image_to_video.enable_model_cpu_offload()
 
-        # temporary enable this if you have some issues with locating the model files
-        setup_debug_logging()
-
         # Initialize Varnish for post-processing
         self.varnish = Varnish(
             device="cuda" if torch.cuda.is_available() else "cpu",