Update app.py

app.py

@@ -189,7 +189,7 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 base = "emilianJR/epiCRealism"
 #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 
-pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
+pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype).to(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
 pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
 pipe.enable_free_init(method="butterworth", use_fast_sampling=False)