Update app.py

app.py

@@ -21,7 +21,8 @@ def run_theia(image):
     mask_generator, sam_model = prepare_mask_generator('cuda')
     depth_anything_model_name = "LiheYoung/depth-anything-large-hf"
     depth_anything_decoder, _ = prepare_depth_decoder(depth_anything_model_name, 'cuda')
-    
+
+    image = torch.transforms.v2.Resize(size=(224, 224))(image)
     images = [image]
     
     theia_decode_results, gt_decode_results = decode_everything(
@@ -44,5 +45,5 @@ def run_theia(image):
 
     return vis_video
 
-demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil", width=224, height=224), outputs=gr.Image(type="numpy"))
+demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil"), outputs=gr.Image(type="numpy"))
 demo.launch()