Update gradio

app.py

@@ -451,9 +451,11 @@ with demo:
 
     with gr.Row():
         encrypted_output_representation = gr.Image(
-            label=f"Encrypted output representation ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", interactive=False
+            label=f"Encrypted output representation ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", 
+            interactive=False,
+            height=256,
+            width=256,
         )
-        encrypted_output_representation.style(height=256, width=256)
 
     gr.Markdown("## Client side")
     gr.Markdown(
@@ -474,13 +476,16 @@ with demo:
             input_image.value,
             label=f"Input image ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):",
             interactive=False,
+            height=256,
+            width=256,   
         )
-        original_image.style(height=256, width=256)
 
         output_image = gr.Image(
-            label=f"Output image ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", interactive=False
+            label=f"Output image ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", 
+            interactive=False,
+            height=256,
+            width=256,
         )
-        output_image.style(height=256, width=256)
 
     # Button to generate the private key
     keygen_button.click(

requirements.txt

@@ -1,2 +1,2 @@
 concrete-ml==1.1.0
-gradio==3.11.0
+gradio==3.40.1