test fix

app.py

@@ -18,7 +18,7 @@ from huggingface_hub import hf_hub_download
 
 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 
-# OPTIONAL: Clear caches at startup to free space (comment out if you prefer not to)
+# OPTIONAL: Clear caches at startup to free space
 hf_cache = os.path.expanduser("~/.cache/huggingface")
 torch_cache = os.path.expanduser("~/.cache/torch")
 if os.path.exists(hf_cache):
@@ -26,6 +26,10 @@ if os.path.exists(hf_cache):
 if os.path.exists(torch_cache):
     shutil.rmtree(torch_cache)
 
+# Configure ZeroGPU to use memory instead of disk
+from spaces.zero.config import Config
+Config.zerogpu_offload_dir = None  # Disable disk offloading to prevent disk space issues
+
 # Load the pipeline with authentication token
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid-xt",
@@ -34,9 +38,6 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
     use_auth_token=os.getenv("HUGGINGFACE_TOKEN")  # Fetch the token from environment if set
 )
 pipe.to("cuda")
-# Uncomment these lines only if you want experimental compilation (and if your environment supports it)
-# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-# pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
 
 max_64_bit_int = 2**63 - 1
 
@@ -145,9 +146,5 @@ with gr.Blocks() as demo:
         api_name="video"
     )
 
-# REMOVED: gr.Examples(...) that referenced local images like "blink_meme.png" which don't exist
-
 if __name__ == "__main__":
-    # remove `share=True` because it is not supported on Hugging Face Spaces
     demo.launch(show_api=False)
-