Allow gif

app.py

@@ -203,8 +203,8 @@ def restore_in_Xmin(
 
     input_format = re.sub(r"^.*\.([^\.]+)$", r"\1", noisy_image)
 
-    if input_format not in ['png', 'webp', 'jpg', 'jpeg', 'bmp']:
-        gr.Warning('Invalid image format. Please first convert into *.png, *.webp, *.jpg, *.jpeg or *.bmp.')
+    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.')
         return None, None, None, None
 
     if output_format == "input":
@@ -229,7 +229,7 @@ def restore_in_Xmin(
         a_prompt = prompt + a_prompt
     print("Final prompt: " + str(a_prompt))
 
-    denoise_image = Image.open(noisy_image if denoise_image is None else denoise_image)
+    denoise_image = np.array(Image.open(noisy_image if denoise_image is None else denoise_image))
 
     if rotation == 90:
         denoise_image = np.array(list(zip(*denoise_image[::-1])))
@@ -241,9 +241,9 @@ def restore_in_Xmin(
 
     if 1 < downscale:
         input_height, input_width, input_channel = denoise_image.shape
-        denoise_image = denoise_image.resize((input_width // downscale, input_height // downscale), Image.LANCZOS)
+        denoise_image = np.array(Image.fromarray(denoise_image).resize((input_width // downscale, input_height // downscale), Image.LANCZOS))
 
-    denoise_image = HWC3(np.array(denoise_image))
+    denoise_image = HWC3(denoise_image)
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
@@ -522,7 +522,7 @@ with gr.Blocks() as interface:
     """)
     gr.HTML(title_html)
 
-    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.jpg, *.bmp)", show_label=True, type="filepath", height=600, elem_id="image-input")
+    input_image = gr.Image(label="Input (*.png, *.webp, *.jpeg, *.jpg, *.gif, *.bmp)", 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)