Update app.py

app.py

@@ -236,8 +236,11 @@ def generate_mask(inp_image, label, threshold):
 
 
 #Setting up Flux (Schnell) Inpainting
-inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
-inpaint_pipe.load_lora_weights("XLabs-AI/flux-RealismLora")
+text_encoder_ = CLIPTextModel.from_pretrained("openai/clip-vit-large-patch14")
+text_encoder_2_ = T5EncoderModel.from_pretrained("xlabs-ai/xflux_text_encoders", torch_dtype=torch.bfloat16)
+
+inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",text_encoder=text_encoder_,text_encoder_2=text_encoder_2_, torch_dtype=torch.bfloat16).to(DEVICE)
+#inpaint_pipe.load_lora_weights("XLabs-AI/flux-RealismLora")
 
 
 #Uncomment the following 4 lines, if you want LoRA Realism weights added to the pipeline