Update

README.md

@@ -4,7 +4,7 @@ emoji: 😻
 colorFrom: purple
 colorTo: red
 sdk: gradio
-sdk_version: 3.36.1
+sdk_version: 4.36.0
 app_file: app.py
 pinned: false
 suggested_hardware: t4-small

app.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import argparse
 import pathlib
 
 import gradio as gr
@@ -44,29 +43,29 @@ def update_slider(choice: str) -> dict:
         "pixar": 121,
         "slamdunk": 119,
     }
-    return gr.Slider.update(maximum=max_vals[choice])
+    return gr.Slider(maximum=max_vals[choice])
 
 
 def update_style_image(style_name: str) -> dict:
     text = get_style_image_markdown_text(style_name)
-    return gr.Markdown.update(value=text)
+    return gr.Markdown(value=text)
 
 
 def set_example_image(example: list) -> dict:
-    return gr.Image.update(value=example[0])
+    return gr.Image(value=example[0])
 
 
 def set_example_styles(example: list) -> list[dict]:
     return [
-        gr.Radio.update(value=example[0]),
-        gr.Slider.update(value=example[1]),
+        gr.Radio(value=example[0]),
+        gr.Slider(value=example[1]),
     ]
 
 
 def set_example_weights(example: list) -> list[dict]:
     return [
-        gr.Slider.update(value=example[0]),
-        gr.Slider.update(value=example[1]),
+        gr.Slider(value=example[0]),
+        gr.Slider(value=example[1]),
     ]
 
 
@@ -75,7 +74,7 @@ model = Model()
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
 
-    with gr.Box():
+    with gr.Group():
         gr.Markdown(
             """## Step 1 (Preprocess Input Image)
 
@@ -99,13 +98,13 @@ with gr.Blocks(css="style.css") as demo:
                     reconstruct_button = gr.Button("Reconstruct Face")
             with gr.Column():
                 reconstructed_face = gr.Image(label="Reconstructed Face", type="numpy")
-                instyle = gr.Variable()
+                instyle = gr.State()
 
         with gr.Row():
             paths = sorted(pathlib.Path("images").glob("*.jpg"))
             gr.Examples(examples=[[path.as_posix()] for path in paths], inputs=input_image)
 
-    with gr.Box():
+    with gr.Group():
         gr.Markdown(
             """## Step 2 (Select Style Image)
 
@@ -115,7 +114,7 @@ with gr.Blocks(css="style.css") as demo:
         )
         with gr.Row():
             with gr.Column():
-                style_type = gr.Radio(label="Style Type", choices=model.style_types)
+                style_type = gr.Radio(label="Style Type", choices=model.style_types, value="cartoon")
                 text = get_style_image_markdown_text("cartoon")
                 style_image = gr.Markdown(value=text)
                 style_index = gr.Slider(label="Style Image Index", minimum=0, maximum=316, step=1, value=26)
@@ -131,7 +130,7 @@ with gr.Blocks(css="style.css") as demo:
                 inputs=[style_type, style_index],
             )
 
-    with gr.Box():
+    with gr.Group():
         gr.Markdown(
             """## Step 3 (Generate Style Transferred Image)
 
@@ -181,4 +180,6 @@ with gr.Blocks(css="style.css") as demo:
         ],
         outputs=result,
     )
-demo.queue(max_size=10).launch()
+
+if __name__ == "__main__":
+    demo.queue(max_size=10).launch()

requirements.txt

@@ -1,7 +1,7 @@
-dlib==19.24.2
-numpy==1.23.5
-opencv-python-headless==4.8.0.74
+dlib==19.24.4
+numpy==1.26.4
+opencv-python-headless==4.10.0.82
 Pillow==9.5.0
-scipy==1.10.1
-torch==2.0.1
-torchvision==0.15.2
+scipy==1.13.1
+torch==2.2.0
+torchvision==0.17.0