new

app.py

@@ -17,12 +17,8 @@ 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",
-    use_auth_token=os.getenv("HUGGINGFACE_TOKEN")  # Fetch the token from the environment variable
+    "stabilityai/stable-video-diffusion-img2vid-xt", torch_dtype=torch.float16, variant="fp16"
 )
 pipe.to("cuda")
 #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
@@ -47,7 +43,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)