Update app.py

app.py

@@ -18,13 +18,12 @@ from torch.multiprocessing import Pool
 #from huggingface_hub import hf_hub_download
 #from safetensors.torch import load_file
 from diffusers import DiffusionPipeline
-from diffusers.utils import load_image
+#from diffusers.utils import load_image
 #import jax
 #import jax.numpy as jnp
 
 def pipe_t2i():
     PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to("cuda")
-    PIPE.unet = torch.compile(PIPE.unet, mode="reduce-overhead", fullgraph=True)
     return PIPE
     
 def translate(text,lang):
@@ -86,11 +85,6 @@ def Piper(name,posi):
     )
     ret1.images[0].save(name)
 
-def tok(pipe,txt):
-    toks = pipe.tokenizer(txt)['input_ids']
-    print(toks)
-    return toks
-
 css="""
 input, input::placeholder {
     text-align: center !important;
@@ -180,7 +174,7 @@ def main():
             with gr.Row():
                 prompt = gr.Textbox(
                     elem_id="prompt",
-                    placeholder="PROMPT",
+                    placeholder="DESCRIPTION",
                     container=False,
                     max_lines=1
                 )