Readd heic format support

app.py

@@ -16,6 +16,9 @@ from gradio_imageslider import ImageSlider
 from PIL import Image
 from SUPIR.util import HWC3, upscale_image, fix_resize, convert_dtype, create_SUPIR_model, load_QF_ckpt
 from huggingface_hub import hf_hub_download
+from pillow_heif import register_heif_opener
+
+register_heif_opener()
 
 max_64_bit_int = 2**32 - 1
 
@@ -203,8 +206,8 @@ def restore_in_Xmin(
 
     input_format = re.sub(r"^.*\.([^\.]+)$", r"\1", noisy_image)
 
-    if input_format not in ['png', 'webp', 'jpg', 'jpeg', 'gif', 'bmp']:
-        gr.Warning('Invalid image format. Please first convert into *.png, *.webp, *.jpg, *.jpeg, *.gif or *.bmp.')
+    if input_format not in ['png', 'webp', 'jpg', 'jpeg', 'gif', 'bmp', 'heic']:
+        gr.Warning('Invalid image format. Please first convert into *.png, *.webp, *.jpg, *.jpeg, *.gif, *.bmp or *.heic.')
         return None, None, None, None
 
     if output_format == "input":
@@ -508,6 +511,7 @@ The images are not stored but the logs are saved during a month.
 ## **How to get SUPIR**
 You can get SUPIR on HuggingFace by [duplicating this space](https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR?duplicate=true) and set GPU.
 You can also install SUPIR on your computer following [this tutorial](https://huggingface.co/blog/MonsterMMORPG/supir-sota-image-upscale-better-than-magnific-ai).
+You can install _Pinokio_ on your computer and then install _SUPIR_ into it. It should be quite easy if you have an Nvidia GPU.
 ## **Terms of use**
 By using this service, users are required to agree to the following terms: The service is a research preview intended for non-commercial use only. It only provides limited safety measures and may generate offensive content. It must not be used for any illegal, harmful, violent, racist, or sexual purposes. The service may collect user dialogue data for future research. Please submit a feedback to us if you get any inappropriate answer! We will collect those to keep improving our models. For an optimal experience, please use desktop computers for this demo, as mobile devices may compromise its quality.
 ## **License**
@@ -526,14 +530,14 @@ with gr.Blocks() as interface:
     """)
     gr.HTML(title_html)
 
-    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.jpg, *.gif, *.bmp)", show_label=True, type="filepath", height=600, elem_id="image-input")
+    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.jpg, *.gif, *.bmp, *.heic)", show_label=True, type="filepath", height=600, elem_id="image-input")
     rotation = gr.Radio([["No rotation", 0], ["⤵ Rotate +90°", 90], ["↩ Return 180°", 180], ["⤴ Rotate -90°", -90]], label="Orientation correction", info="Will apply the following rotation before restoring the image; the AI needs a good orientation to understand the content", value=0, interactive=True, visible=False)
     with gr.Group():
         prompt = gr.Textbox(label="Image description", info="Help the AI understand what the image represents; describe as much as possible, especially the details we can't see on the original image; you can write in any language", value="", placeholder="A 33 years old man, walking, in the street, Santiago, morning, Summer, photorealistic", lines=3)
         prompt_hint = gr.HTML("You can use a <a href='"'https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b'"'>LlaVa space</a> to auto-generate the description of your image.")
         upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8], ["x9", 9], ["x10", 10]], label="Upscale factor", info="Resolution x1 to x10", value=2, interactive=True)
         allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min (discouraged)", 8], ["9 min (discouraged)", 9], ["10 min (discouraged)", 10]], label="GPU allocation time", info="lower=May abort run, higher=Quota penalty for next runs", value=7, interactive=True)
-        output_format = gr.Radio([["As input", "input"], ["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="input", interactive=True)
+        output_format = gr.Radio([["As input", "input"], ["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"], ["*.heic", "heic"]], label="Image format for result", info="File extention", value="input", interactive=True)
 
     with gr.Accordion("Pre-denoising (optional)", open=False):
         gamma_correction = gr.Slider(label="Gamma Correction", info = "lower=lighter, higher=darker", minimum=0.1, maximum=2.0, value=1.0, step=0.1)

requirements.txt

@@ -1,41 +1,42 @@
-fastapi>=0.111.0
-gradio>=4.31.4
-gradio_imageslider>=0.0.20
-gradio_client>=0.16.4
-Markdown>=3.6
-numpy>=1.26.4
-requests>=2.32.2
-sentencepiece>=0.2.0
-tokenizers>=0.11.3
-torchvision>=0.18.0
-uvicorn>=0.29.0
-wandb>=0.17.0
-httpx>=0.27.0
-transformers>=4.41.0
-accelerate>=0.30.1
-scikit-learn>=1.5.0
-sentencepiece>=0.2.0
-einops>=0.8.0
-einops-exts>=0.0.4
-timm>=1.0.3
-openai-clip>=1.0.1
-fsspec>=2024.5.0
-kornia>=0.7.2
-matplotlib>=3.9.0
-ninja>=1.11.1.1
-omegaconf>=2.3.0
-open-clip-torch==2.24.0
-opencv-python>=4.9.0.80
-pandas>=2.2.2
-Pillow>=10.3.0
-pytorch-lightning>=2.2.4
-PyYAML>=6.0.1
-scipy>=1.13.0
-tqdm>=4.66.4
-triton>=2.3.0
-urllib3>=2.2.1
-webdataset>=0.2.86
-xformers>=0.0.26.post1
-facexlib>=0.3.0
-k-diffusion>=0.1.1.post1
-diffusers>=0.27.2
+fastapi>=0.111.0
+gradio>=4.31.4
+gradio_imageslider>=0.0.20
+gradio_client>=0.16.4
+Markdown>=3.6
+numpy>=1.26.4
+requests>=2.32.2
+sentencepiece>=0.2.0
+tokenizers>=0.11.3
+torchvision>=0.18.0
+uvicorn>=0.29.0
+wandb>=0.17.0
+httpx>=0.27.0
+transformers>=4.41.0
+accelerate>=0.30.1
+scikit-learn>=1.5.0
+sentencepiece>=0.2.0
+einops>=0.8.0
+einops-exts>=0.0.4
+timm>=1.0.3
+openai-clip>=1.0.1
+fsspec>=2024.5.0
+kornia>=0.7.2
+matplotlib>=3.9.0
+ninja>=1.11.1.1
+omegaconf>=2.3.0
+open-clip-torch==2.24.0
+opencv-python>=4.9.0.80
+pandas>=2.2.2
+Pillow>=10.3.0
+pytorch-lightning>=2.2.4
+PyYAML>=6.0.1
+scipy>=1.13.0
+tqdm>=4.66.4
+triton>=2.3.0
+urllib3>=2.2.1
+webdataset>=0.2.86
+xformers>=0.0.26.post1
+facexlib>=0.3.0
+k-diffusion>=0.1.1.post1
+diffusers>=0.27.2
+pillow-heif>=0.17.0