Update app.py

app.py

@@ -16,7 +16,7 @@ from diffusers import DiffusionPipeline, FlaxStableDiffusionPipeline
 
 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", 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)
 
 def translate(text,lang):