Update app.py

app.py

@@ -8,7 +8,7 @@ from diffusers import DiffusionPipeline
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16, revision="refs/pr/1").to(device)
+pipe = DiffusionPipeline.from_pretrained("Niansuh/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -45,7 +45,7 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""# FLUX.1 [schnell]
-12B param rectified flow transformer distilled from [FLUX.1 [pro]] for 4 step generation
+12B param rectified flow transformer distilled from [FLUX.1 pro] for 4 step generation
 [[model](https://huggingface.co/Niansuh/FLUX.1-schnell)]
         """)
         
@@ -119,4 +119,3 @@ with gr.Blocks(css=css) as demo:
         outputs = [result, seed]
     )
 
-demo.launch()