Update app.py

app.py

@@ -10,7 +10,7 @@ import numpy as np
 from lxml.html import fromstring
 from pathos.threading import ThreadPool as Pool
 from diffusers import DiffusionPipeline
-#from diffusers.pipelines.stable_diffusion.pipeline_flax_stable_diffusion import FlaxStableDiffusionPipeline as FluxPipeline
+from diffusers.pipelines.stable_diffusion.pipeline_flax_stable_diffusion import FlaxStableDiffusionPipeline as FluxPipeline
 #from diffusers.utils import export_to_gif
 #from huggingface_hub import hf_hub_download
 #from safetensors.torch import load_file
@@ -18,6 +18,7 @@ from diffusers import DiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
+pipe_flx = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
 
 def translate(text,lang):
     
@@ -82,7 +83,7 @@ def Piper(_do,_dont):
     try:
         neg = pipe.tokenizer(_dont)['input_ids']
         print(neg)
-        retu = pipe(
+        retu = pipe_flx(
             _do,
             neg_prompt_ids=neg,
             height=512,