Update

README.md

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

app.py

@@ -92,7 +92,7 @@ path = "controller_age015id025exp02hai04ori02gam15/"
 controller = Controller(path, device)
 fn = functools.partial(run, controller=controller, device=device)
 
-gr.Interface(
+demo = gr.Interface(
     fn=fn,
     inputs=[
         gr.Slider(label="Seed", minimum=0, maximum=1000000, step=1, value=0),
@@ -107,11 +107,14 @@ gr.Interface(
         gr.Slider(label="Number of Columns", minimum=1, maximum=5, step=1, value=5),
     ],
     outputs=[
-        gr.Image(label="Generated Image", type="pil"),
-        gr.Image(label="Head Pose Controlled", type="pil"),
-        gr.Image(label="Age Controlled", type="pil"),
-        gr.Image(label="Hair Color Controlled", type="pil"),
+        gr.Image(label="Generated Image"),
+        gr.Image(label="Head Pose Controlled"),
+        gr.Image(label="Age Controlled"),
+        gr.Image(label="Hair Color Controlled"),
     ],
     title=TITLE,
     description=DESCRIPTION,
-).queue(max_size=10).launch()
+)
+
+if __name__ == "__main__":
+    demo.queue(max_size=10).launch()

requirements.txt

@@ -1,4 +1,4 @@
-numpy==1.23.5
-Pillow==10.0.0
-torch==1.11.0
-torchvision==0.12.0
+numpy==1.26.4
+Pillow==10.3.0
+torch==2.0.1
+torchvision==0.15.2