Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import gradio.helpers
 import torch
 import os
 from glob import glob
@@ -9,11 +10,12 @@ from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 from PIL import Image
 
-
 import uuid
 import random
 from huggingface_hub import hf_hub_download
 
+gradio.helpers.CACHED_FOLDER = '/data/cache'
+
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid-xt", torch_dtype=torch.float16, variant="fp16"
 )