Update app.py

app.py

@@ -98,11 +98,13 @@ else:
     os.system(f'echo "Installing Deforum animation extension..."')
     os.system(f"git clone https://github.com/deforum-art/deforum-for-automatic1111-webui  /home/user/app/stable-diffusion-webui/extensions/deforum-for-automatic1111-webui")
     
-    # Load Protogen model (example custom model)
-    os.system(f'echo "Downloading OpenGen v1.0 Stable Diffusion model..."')
-    os.system(f"wget -q https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors  -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/OpenGen_v1.0.safetensors")
-    
-    # Critical dependency installation sequence
+   # Load Protogen model (example custom model)
+os.system(f'echo "Downloading OpenGen v1.0 Stable Diffusion model..."')
+os.system(f"wget -q https://huggingface.co/darkstorm2150/OpenGen/resolve/main/OpenGen%20v1.0.safetensors  -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/OpenGen_v1.0.safetensors")
+
+# Load Flux model
+os.system(f'echo "Downloading Flux model from Civitai..."')
+os.system(f'wget -q "https://civitai.com/api/download/models/691639?type=Model&format=SafeTensor&size=pruned&fp=fp32" -O /home/user/app/stable-diffusion-webui/models/Stable-diffusion/flux.safetensors')
     #os.system(f'echo "starting dependency cleanup..."')
     #os.system(f'pip cache purge')