api_key

app.py

@@ -17,8 +17,12 @@ from huggingface_hub import hf_hub_download
 
 #gradio.helpers.CACHED_FOLDER = '/data/cache'
 
+# Load the pipeline with authentication token
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "multimodalart/stable-video-diffusion", torch_dtype=torch.float16, variant="fp16"
+    "multimodalart/stable-video-diffusion",
+    torch_dtype=torch.float16,
+    variant="fp16",
+    use_auth_token=os.getenv("HUGGINGFACE_TOKEN")  # Fetch the token from the environment variable
 )
 pipe.to("cuda")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
@@ -43,7 +47,7 @@ def sample(
     if image.mode == "RGBA":
         image = image.convert("RGB")
         
-    if(randomize_seed):
+    if randomize_seed:
         seed = random.randint(0, max_64_bit_int)
     generator = torch.manual_seed(seed)
     
@@ -125,4 +129,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     #demo.queue(max_size=20, api_open=False)
-    demo.launch(share=True, show_api=False)
+    demo.launch(share=True, show_api=False)