Update app.py

app.py

@@ -32,7 +32,7 @@ def init_pool():
 _dynamo.config.suppress_errors = True
 mp.set_start_method("fork", force=True)
 
-#pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1", token=os.getenv("hf_token")).to(device)
+#pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token")).to(device)
 #pipe2 = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to(device)
 #pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
 
@@ -41,7 +41,7 @@ PIPE = None
 def pipe_t2i():
     global PIPE
     if PIPE is None:
-        PIPE = pipeline("text-to-image", model="black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, tokenizer="black-forest-labs/FLUX.1-schnell", device=-1, token=os.getenv("hf_token"))
+        PIPE = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, token=os.getenv("hf_token"), device=-1)
     return PIPE
     
 def pipe_i2i():