Update app.py

app.py

@@ -1,3 +1,8 @@
+"""
+Some modified code included from:
+- https://github.com/nidhaloff/deep-translator
+"""
+
 import urllib
 import requests
 from bs4 import BeautifulSoup
@@ -33,7 +38,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, AutoencoderTiny, AutoencoderKL
+from diffusers import DiffusionPipeline
 from PIL import Image, ImageDraw, ImageFont
 from transformers import pipeline, T5ForConditionalGeneration, T5Tokenizer
 from refiners.fluxion.utils import manual_seed
@@ -502,8 +507,6 @@ function custom(){
 
 # torch pipes
 
-taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
-good_vae = AutoencoderKL.from_pretrained("ostris/Flex.1-alpha", subfolder="vae", torch_dtype=dtype).to(device)
 image_pipe = DiffusionPipeline.from_pretrained("ostris/Flex.1-alpha", torch_dtype=dtype, vae=taef1).to(device)
 image_pipe.enable_model_cpu_offload()
 image_pipe.enable_vae_slicing()
@@ -621,7 +624,6 @@ def pipe_generate_image(p1,p2):
             num_images_per_prompt=1,
             num_inference_steps=image_steps,
             max_sequence_length=seq,
-            good_vae=good_vae,
             generator=torch.Generator(device).manual_seed(random.randint(0, MAX_SEED))
     ).images
     log(f'RET pipe_generate')