Update

README.md

@@ -4,7 +4,7 @@ emoji: ⚡
 colorFrom: indigo
 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

@@ -22,13 +22,13 @@ def load_hairstyle_list() -> list[str]:
 
 
 def set_example_image(example: list) -> dict:
-    return gr.Image.update(value=example[0])
+    return gr.Image(value=example[0])
 
 
 def update_step2_components(choice: str) -> tuple[dict, dict]:
     return (
-        gr.Dropdown.update(visible=choice in ["hairstyle", "both"]),
-        gr.Textbox.update(visible=choice in ["color", "both"]),
+        gr.Dropdown(visible=choice in ["hairstyle", "both"]),
+        gr.Textbox(visible=choice in ["color", "both"]),
     )
 
 
@@ -36,7 +36,7 @@ model = Model()
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)
-    with gr.Box():
+    with gr.Group():
         gr.Markdown("## Step 1")
         with gr.Row():
             with gr.Column():
@@ -48,13 +48,13 @@ with gr.Blocks(css="style.css") as demo:
                 aligned_face = gr.Image(label="Aligned Face", type="pil", interactive=False)
             with gr.Column():
                 reconstructed_face = gr.Image(label="Reconstructed Face", type="numpy")
-                latent = gr.Variable()
+                latent = 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")
         with gr.Row():
             with gr.Column():
@@ -87,4 +87,5 @@ 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,8 +1,8 @@
-dlib==19.23.0
+dlib==19.24.4
 git+https://github.com/openai/CLIP.git
-numpy==1.22.3
-opencv-python-headless==4.5.5.64
-Pillow==9.1.0
-scipy==1.8.0
-torch==1.11.0
-torchvision==0.12.0
+numpy==1.26.4
+opencv-python-headless==4.10.0.82
+Pillow==9.5.0
+scipy==1.13.1
+torch==2.0.1
+torchvision==0.15.2