Update

README.md

@@ -4,8 +4,7 @@ emoji: 🐢
 colorFrom: blue
 colorTo: pink
 sdk: gradio
-sdk_version: 3.36.1
-python_version: 3.10.11
+sdk_version: 4.36.1
 app_file: app.py
 pinned: false
 suggested_hardware: a10g-small

app_generated_image.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import os
 import pathlib
 import shlex
 import subprocess
@@ -122,7 +121,7 @@ def process_example(source_prompt: str, seed: int, translation_prompt: str) -> t
 
 def create_prompt_demo() -> gr.Blocks:
     with gr.Blocks() as demo:
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("Step 1 (This step will take about 1.5 minutes on A10G.)")
             with gr.Row():
                 with gr.Column():
@@ -137,7 +136,7 @@ def create_prompt_demo() -> gr.Blocks:
                 with gr.Column():
                     generated_image = gr.Image(label="Generated image", type="filepath")
                     exp_name = gr.Text(visible=False)
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("Step 2 (This step will take about 1.5 minutes on A10G.)")
             with gr.Row():
                 with gr.Column():
@@ -177,7 +176,6 @@ def create_prompt_demo() -> gr.Blocks:
                     result,
                 ],
                 fn=process_example,
-                cache_examples=os.getenv("CACHE_EXAMPLES"),
             )
 
         extract_feature_button.click(

app_real_image.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import hashlib
-import os
 import pathlib
 import shlex
 import subprocess
@@ -109,7 +108,7 @@ def process_example(image: str, translation_prompt: str, negative_prompt: str) -
 
 def create_real_image_demo():
     with gr.Blocks() as demo:
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("Step 1 (This step will take about 5 minutes on A10G.)")
             with gr.Row():
                 with gr.Column():
@@ -118,7 +117,7 @@ def create_real_image_demo():
                 with gr.Column():
                     reconstructed_image = gr.Image(label="Reconstructed image", type="filepath")
                     exp_name = gr.Text(visible=False)
-        with gr.Box():
+        with gr.Group():
             gr.Markdown("Step 2 (This step will take about 1.5 minutes on A10G.)")
             with gr.Row():
                 with gr.Column():
@@ -172,7 +171,6 @@ def create_real_image_demo():
                     result,
                 ],
                 fn=process_example,
-                cache_examples=os.getenv("CACHE_EXAMPLES"),
             )
 
         extract_feature_button.click(

requirements.txt

@@ -2,8 +2,8 @@ albumentations==1.3.0
 -e git+https://github.com/openai/CLIP.git@main#egg=clip
 diffusers==0.12.1
 einops==0.6.0
-gradio==3.36.1
-huggingface-hub==0.16.4
+gradio==4.36.1
+huggingface-hub==0.23.3
 imageio==2.25.0
 imageio-ffmpeg==0.4.8
 kornia==0.6.9