Update app.py

app.py

@@ -206,6 +206,7 @@ def main():
 
     progress=gr.Progress()
 
+    base = "black-forest-labs/FLUX.1-dev"
     #base="SG161222/Realistic_Vision_V6.0_B1_noVAE"
     #vae="stabilityai/sd-vae-ft-mse-original"
     #repo = "ByteDance/SDXL-Lightning"
@@ -214,7 +215,7 @@ def main():
     #unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device)
     #unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
 
-    base = "emilianJR/epiCRealism"
+    #base = "emilianJR/epiCRealism"
     repo = "ByteDance/AnimateDiff-Lightning"
     ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"