ebinan92 commited on
Commit
5de4d16
·
verified ·
1 Parent(s): f12e229

Fix PIL fp closed error on image preview (eagerly load pixel data)

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -140,8 +140,9 @@ def _load_image(file_path: str, page_number: int) -> tuple[int, Image.Image]:
140
  if path.suffix.lower() == ".pdf":
141
  return _render_pdf_page(path, page_number)
142
 
143
- with Image.open(path) as img:
144
- return 1, _resize_if_needed(img)
 
145
 
146
 
147
  @spaces.GPU(duration=180)
 
140
  if path.suffix.lower() == ".pdf":
141
  return _render_pdf_page(path, page_number)
142
 
143
+ img = Image.open(path)
144
+ img.load() # eagerly read pixel data so the fp can be released
145
+ return 1, _resize_if_needed(img)
146
 
147
 
148
  @spaces.GPU(duration=180)