Update app.py

app.py

@@ -1,7 +1,6 @@
 """
 Modified parts included from these sources:
 - https://github.com/nidhaloff/deep-translator
-- https://huggingface.co/spaces/ostris/Flex.1-alpha
 """
 
 import urllib
@@ -514,14 +513,11 @@ 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 = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, vae=taef1).to(device) # ostris/Flex.1-alpha
+image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, vae=taef1).to(device)
 #image_pipe.enable_model_cpu_offload()
 
 torch.cuda.empty_cache()
 
-#image_pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(image_pipe)
-
 # functionality
 
 @spaces.GPU(duration=300)