multimodalart HF staff commited on
Commit
6c51e38
1 Parent(s): daf9b1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,7 +2,9 @@ import gradio as gr
2
  import torch
3
  import spaces
4
  from diffusers import FluxInpaintPipeline
5
- from PIL import Image
 
 
6
 
7
  # Initialize the pipeline
8
  pipe = FluxInpaintPipeline.from_pretrained(
 
2
  import torch
3
  import spaces
4
  from diffusers import FluxInpaintPipeline
5
+ from PIL import Image, ImageFile
6
+
7
+ ImageFile.LOAD_TRUNCATED_IMAGES = True
8
 
9
  # Initialize the pipeline
10
  pipe = FluxInpaintPipeline.from_pretrained(