jiuface commited on
Commit
f6e14e0
·
verified ·
1 Parent(s): a1947de

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +4 -1
src/app.py CHANGED
@@ -2,7 +2,7 @@ import spaces
2
  import torch
3
  from io import BytesIO
4
  import PIL.Image
5
-
6
 
7
  from pathlib import Path
8
 
@@ -16,6 +16,9 @@ import requests
16
  from enhancer import ESRGANUpscaler, ESRGANUpscalerCheckpoints
17
  import time
18
 
 
 
 
19
  TITLE = """
20
  Image Enhancer
21
  """
 
2
  import torch
3
  from io import BytesIO
4
  import PIL.Image
5
+ import pillow_heif
6
 
7
  from pathlib import Path
8
 
 
16
  from enhancer import ESRGANUpscaler, ESRGANUpscalerCheckpoints
17
  import time
18
 
19
+ pillow_heif.register_heif_opener()
20
+ pillow_heif.register_avif_opener()
21
+
22
  TITLE = """
23
  Image Enhancer
24
  """