Update app.py

app.py

@@ -1,9 +1,13 @@
 import os
+# Set cache directories to writable locations right at the beginning
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
+
 import gc
 import torch
 import cv2
 import gradio as gr
-import gradio as gr
 print("📦 Gradio version:", gr.__version__)
 import numpy as np
 import matplotlib.cm as cm
@@ -51,18 +55,12 @@ model_name = encoder2name[encoder]
 # Initialize the model.
 video_depth_anything = VideoDepthAnything(**model_configs[encoder])
 
-# Set cache directories to writable locations
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
-os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
-
 filepath = hf_hub_download(
     repo_id=f"depth-anything/Video-Depth-Anything-{model_name}",
     filename=f"video_depth_anything_{encoder}.pth",
     repo_type="model",
     cache_dir="/tmp/huggingface"  # Explizites Setzen des Cache-Verzeichnisses
 )
-
 video_depth_anything.load_state_dict(torch.load(filepath, map_location='cpu'))
 video_depth_anything = video_depth_anything.to(DEVICE).eval()
 
@@ -177,9 +175,10 @@ def construct_demo():
                 input_video = gr.Video(label="Input Video")
             with gr.Column(scale=2):
                 with gr.Row(equal_height=True):
-                    processed_video = gr.Video(label="Preprocessed Video", interactive=False, autoplay=True, loop=True, show_share_button=True, scale=5)
-                    depth_vis_video = gr.Video(label="Generated Depth Video", interactive=False, autoplay=True, loop=True, show_share_button=True, scale=5)
-                    stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, loop=True, show_share_button=True, scale=5)
+                    # Removed loop parameter which is not supported in Gradio 4.36.0
+                    processed_video = gr.Video(label="Preprocessed Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
+                    depth_vis_video = gr.Video(label="Generated Depth Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
+                    stitched_video = gr.Video(label="Stitched RGBD Video", interactive=False, autoplay=True, show_share_button=True, scale=5)
                     
         with gr.Row(equal_height=True):
             with gr.Column(scale=1):