Update app.py

app.py

@@ -1,24 +1,22 @@
 import gradio as gr
 import torch
-from diffusers import AutoencoderKL, FluxTransformer2DModel, FluxFillPipeline
+from diffusers import FluxFillPipeline
 from diffusers.utils import load_image
-from controlnet_flux import FluxControlNetModel
-from transformer_flux import FluxTransformer2DModel
-from pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
-from transformers import T5EncoderModel, CLIPTextModel
+#from controlnet_flux import FluxControlNetModel
+#from transformer_flux import FluxTransformer2DModel
+#from pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
+#from transformers import T5EncoderModel, CLIPTextModel
 from PIL import Image, ImageDraw
 import numpy as np
 import spaces
 from huggingface_hub import hf_hub_download
-from optimum.quanto import freeze, qfloat8, quantize
+#from optimum.quanto import freeze, qfloat8, quantize
 
 pipe = FluxFillPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-Fill-dev",
     torch_dtype=torch.bfloat16
 ).to("cuda")
 
-pipe.to("cuda")
-
 def can_expand(source_width, source_height, target_width, target_height, alignment):
     if alignment in ("Left", "Right") and source_width >= target_width:
         return False