update dependenices and fix deprecation warnings

README.md

@@ -4,10 +4,21 @@ emoji: 👁
 colorFrom: blue
 colorTo: red
 sdk: gradio
-sdk_version: 2.9.1
+sdk_version: 3.17.1
 app_file: app.py
 pinned: false
 license: mit
+python_version: 3.8.9
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+Photos stylisation using deep neural networks: anime and art style.
+
+Read more at [7bits.it/portfolio/nijigenka-ai](https://7bits.it/portfolio/nijigenka-ai).
+
+## Local development
+
+Install gradio `pip install gradio==3.17.1`.
+
+Install other deps: `pip install -r requirements.txt`.
+
+Run application `python app.py --models_repo_id=/path/to/dir/with/models`. You can ommit `--models_repo_id` if you would like to download models from the hugging face hub.

app.py

@@ -1,5 +1,4 @@
 import argparse
-from cProfile import label
 import functools
 import pathlib
 import os
@@ -42,6 +41,7 @@ def join_image_h(im1: PIL.Image.Image, im2: PIL.Image.Image) -> PIL.Image.Image:
 def predict(
     image: PIL.Image.Image,
     style: str,
+    *,
     face_aligner: FaceAligner,
     encoder: Encoder,
     generator: Dict[str, Generator],
@@ -65,6 +65,7 @@ def predict(
 def get_model_path(repo_id: str, filename: str):
     maybe_path = os.path.join(repo_id, filename)
     if os.path.exists(maybe_path):
+        print('Using local models')
         return os.path.abspath(maybe_path)
     else:
         return hf_hub_download(
@@ -120,20 +121,20 @@ def main():
     iface = gr.Interface(
         fn=func,
         inputs=[
-            gr.inputs.Image(
+            gr.Image(
                 type='pil',
                 label='Real photo with a face',
             ),
-            gr.inputs.Radio(
+            gr.Radio(
                 choices=generator_types,
                 type='value',
-                default=generator_types[0],
+                value=generator_types[0],
                 label='Style',
             ),
         ],
         outputs=[
-            gr.outputs.Carousel(['image'], label='Result'),
-            gr.outputs.Textbox(type="auto", label='Error'),
+            gr.Gallery(label='Result'),
+            gr.Textbox(label='Error'),
         ],
         examples=load_examples(),
         title='Nijigenka: Portrait to Art',

generator.py

@@ -1,4 +1,3 @@
-from xml.etree.ElementTree import PI
 import PIL.Image
 import onnxruntime
 import numpy as np

requirements.txt

@@ -1,6 +1,9 @@
-huggingface-hub==0.4.0
-pillow==8.4.0
-numpy==1.21.4
-onnxruntime==1.10.0
-dlib==19.23.1
-scipy==1.7.3
+huggingface-hub==0.12.0
+pillow==9.4.0
+numpy==1.24.1
+onnxruntime==1.13.1
+dlib==19.24.0
+scipy==1.10.0
+
+# for local dev
+# gradio==3.17.1