Update app.py

app.py

@@ -13,7 +13,7 @@ vae = AutoencoderTiny.from_pretrained(model_id, safetensors=True).to(device)
 
 @torch.no_grad()
 def encode(image):
-    DIM = 512
+    DIM = [512]
     processed = center_crop(resize(image, DIM), DIM)
     tensor = to_tensor(processed).unsqueeze(0).to(device)
     latents = vae.encoder(tensor)
@@ -27,7 +27,6 @@ astronaut = os.path.join(os.path.dirname(__file__), "images/6.png")
 def app():
     return gr.Interface(encode,
                         gr.Image(type="pil",
-                                 mirror_webcam=False,
                                  label='512x512',
                                  value=astronaut),
                         gr.Image(type="pil",
@@ -40,7 +39,7 @@ def app():
                             astronaut,
                             os.path.join(os.path.dirname(__file__), "images/7.png"),
                             os.path.join(os.path.dirname(__file__), "images/34.png")
-                        ], allow_flagging='never', title='Image Encoder')
+                        ], flagging_mode='never', title='Image Encoder')
 
 
 if __name__ == "__main__":