Update app.py

app.py

@@ -9,14 +9,14 @@ import gradio as gr
 import numpy as np
 from lxml.html import fromstring
 from pathos.threading import ThreadPool as Pool
-from diffusers import DiffusionPipeline, FlaxStableDiffusionPipeline
+#from diffusers import DiffusionPipeline,
+from diffusers.pipelines.flux import FluxPipeline
 #from diffusers.utils import export_to_gif
 #from huggingface_hub import hf_hub_download
 #from safetensors.torch import load_file
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device)
-pipe_flx = FlaxStableDiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", token=os.getenv("hf_token")).to(device)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device)
 
 def translate(text,lang):
     
@@ -79,7 +79,7 @@ def generate_random_string(length):
 @spaces.GPU(duration=35)
 def Piper(_do,neg):
     try:
-        retu = pipe_flx(
+        retu = pipe(
             _do,
             neg_prompt_ids=neg,
             height=512,