Update app.py

app.py

@@ -33,7 +33,7 @@ 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 DiffusionPipeline
+from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
 from PIL import Image, ImageDraw, ImageFont
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
 from refiners.fluxion.utils import manual_seed
@@ -616,7 +616,6 @@ def pipe_generate_image(p1,p2):
             progress=gr.Progress(track_tqdm=True),
             height=height,
             width=width,
-            safety_checker=None,
             guidance_scale=img_accu,
             num_images_per_prompt=1,
             num_inference_steps=image_steps,