Update app.py

app.py

@@ -1,7 +1,7 @@
 import torch
 import spaces
 import gradio as gr
-from diffusers import FluxInpaintPipeline
+from diffusers import FluxInpaintPipeline, FluxTransformer2DModel
 import random
 import numpy as np
 
@@ -13,13 +13,10 @@ if torch.cuda.is_available():
         "https://huggingface.co/black-forest-labs/FLUX.1-Fill-dev/blob/main/flux1-fill-dev.safetensors",
         torch_dtype=torch.bfloat16
     )
-    pipe = FluxPipeline.from_pretrained(
+    pipe = FluxInpaintPipeline.from_pretrained(
         model, 
         transformer=transformer,
         torch_dtype=torch.bfloat16)
-    pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_config(
-        pipe.scheduler.config, use_beta_sigmas=True
-    )
     pipe.to("cuda")