Update app.py

app.py

@@ -24,7 +24,7 @@ from lxml.html import fromstring
 from diffusers.utils import export_to_video, load_image
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
-from diffusers import StableDiffusionPipeline, CogVideoXImageToVideoPipeline
+from diffusers import FluxPipeline, CogVideoXImageToVideoPipeline
 from PIL import Image, ImageDraw, ImageFont
 
 # logging
@@ -47,7 +47,7 @@ else:
     device = "cpu"
     dtype = torch.float32
 
-base = "emilianJR/epiCRealism"
+base = "black-forest-labs/FLUX.1-dev"
 
 # variable data
 
@@ -107,7 +107,7 @@ function custom(){
 
 # torch pipes
 
-image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
+image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
 video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
     "THUDM/CogVideoX-5b-I2V",
     torch_dtype=dtype