Update app.py

app.py

@@ -38,9 +38,9 @@ import gradio as gr
 from lxml.html import fromstring
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file, save_file
-from diffusers import FluxPipeline, AutoencoderTiny, AutoencoderKL, FluxPipeline, FlowMatchEulerDiscreteScheduler
+from diffusers import StableDiffusionPipeline
 from PIL import Image, ImageDraw, ImageFont
-from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer, CLIPTextModel, CLIPTokenizer, T5EncoderModel
+from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
 from refiners.fluxion.utils import manual_seed
 from refiners.foundationals.latent_diffusion import Solver, solvers
 from refiners.foundationals.latent_diffusion.stable_diffusion_1.multi_upscaler import (
@@ -512,7 +512,7 @@ function custom(){
 
 # torch pipes
 
-image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
+image_pipe = StableDiffusionPipeline.from_pretrained("emilianJR/epiCRealism", torch_dtype=dtype).to(device)
 image_pipe.enable_model_cpu_offload()
 
 torch.cuda.empty_cache()