Handles HEIC

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 import numpy as np
-import os
 import time
 import math
 import random
-import imageio
 import torch
 import spaces
 
@@ -13,6 +11,9 @@ from diffusers import (
     StableDiffusionControlNetPipeline,
 )
 from PIL import Image
+from pillow_heif import register_heif_opener
+
+register_heif_opener()
 
 max_64_bit_int = np.iinfo(np.int32).max
 
@@ -177,10 +178,13 @@ with gr.Blocks() as interface:
         input_image = gr.Image(label = "Your image", sources = ["upload", "webcam", "clipboard"], type = "pil")
         prompt = gr.Textbox(label = "Prompt", info = "Instruct what to change in the image", placeholder = "Order the AI what to change in the image", lines = 2)
         with gr.Accordion("Advanced options", open = False):
-            negative_prompt = gr.Textbox(label = "Negative prompt", placeholder = "Describe what you do NOT want to see in the image", value = "Watermark")
+            negative_prompt = gr.Textbox(label = "Negative prompt", placeholder = "Describe what you do NOT want to see in the image", value = ''
+                                    'blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, '
+                                    'worst quality, low quality, frames, watermark, signature, jpeg artifacts, '
+                                    'deformed, lowres, over-smooth')
             denoising_steps = gr.Slider(minimum = 0, maximum = 1000, value = 0, step = 1, label = "Denoising", info = "lower=irrelevant result, higher=relevant result")
             num_inference_steps = gr.Slider(minimum = 10, maximum = 500, value = 20, step = 1, label = "Number of inference steps", info = "lower=faster, higher=image quality")
-            guidance_scale = gr.Slider(minimum = 1, maximum = 13, value = 5, step = 0.1, label = "Classifier-Free Guidance Scale", info = "lower=image quality, higher=follow the prompt")
+            guidance_scale = gr.Slider(minimum = 1, maximum = 13, value = 5, step = 0.1, label = "Guidance Scale", info = "lower=image quality, higher=follow the prompt")
             image_guidance_scale = gr.Slider(minimum = 1, value = 1.5, step = 0.1, label = "Image Guidance Scale", info = "lower=image quality, higher=follow the image")
             randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different")
             seed = gr.Slider(minimum = 0, maximum = max_64_bit_int, step = 1, randomize = True, label = "Seed")
@@ -242,7 +246,9 @@ with gr.Blocks() as interface:
                 [
                     "./Examples/Example1.webp",
                     "What if it's snowing?",
-                    "Watermark",
+                    "blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, "
+                        "worst quality, low quality, frames, watermark, signature, jpeg artifacts, "
+                        "deformed, lowres, over-smooth",
                     1,
                     20,
                     5,
@@ -253,7 +259,9 @@ with gr.Blocks() as interface:
                 [
                     "./Examples/Example2.png",
                     "What if this woman had brown hair?",
-                    "Watermark",
+                    "blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, "
+                        "worst quality, low quality, frames, watermark, signature, jpeg artifacts, "
+                        "deformed, lowres, over-smooth",
                     1,
                     20,
                     5,
@@ -264,7 +272,9 @@ with gr.Blocks() as interface:
                 [
                     "./Examples/Example3.jpeg",
                     "Replace the house by a windmill",
-                    "Watermark",
+                    "blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, "
+                        "worst quality, low quality, frames, watermark, signature, jpeg artifacts, "
+                        "deformed, lowres, over-smooth",
                     1,
                     20,
                     5,
@@ -275,7 +285,9 @@ with gr.Blocks() as interface:
                 [
                     "./Examples/Example4.gif",
                     "What if the camera was in opposite side?",
-                    "Watermark",
+                    "blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, "
+                        "worst quality, low quality, frames, watermark, signature, jpeg artifacts, "
+                        "deformed, lowres, over-smooth",
                     1,
                     20,
                     5,
@@ -286,7 +298,9 @@ with gr.Blocks() as interface:
                 [
                     "./Examples/Example5.bmp",
                     "Turn him into cyborg",
-                    "Watermark",
+                    "blurring, aliasing, unsharp, weird textures, ugly, dirty, messy, "
+                        "worst quality, low quality, frames, watermark, signature, jpeg artifacts, "
+                        "deformed, lowres, over-smooth",
                     1,
                     20,
                     5,