Use image format mapped by an extension dictionary, which requires Pi…

…llow 9.4.0 to work properly.

fooocusapi/utils/file_utils.py

@@ -51,6 +51,7 @@ def save_output_file(
 
     if extension not in ['png', 'jpg', 'webp']:
         extension = 'png'
+    image_format = Image.registered_extensions()['.'+extension]
 
     if image_meta is None:
         image_meta = {}
@@ -64,7 +65,7 @@ def save_output_file(
     os.makedirs(os.path.dirname(file_path), exist_ok=True)
     Image.fromarray(img).save(
         file_path,
-        format=extension,
+        format=image_format,
         pnginfo=meta,
         optimize=True)
     return Path(filename).as_posix()

requirements.txt

@@ -4,7 +4,7 @@ transformers==4.30.2
 safetensors==0.3.1
 accelerate==0.21.0
 pyyaml==6.0
-Pillow==9.2.0
+Pillow==9.4.0
 scipy==1.9.3
 tqdm==4.64.1
 psutil==5.9.5