a couple major fixies!

Dockerfile

@@ -1,5 +1,5 @@
 # Use an official PyTorch image with CUDA support as the base image
-FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
 
 # Set the working directory
 WORKDIR /app

main.py

@@ -37,7 +37,7 @@ async def upload_zip(file: UploadFile = File(...)):
         os.makedirs(output_dir, exist_ok=True)
 
         # Define the model weights path
-        weights_path = "/path/to/your/model/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth"
+        weights_path = "/app/dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth"
 
         # # Check if CUDA is available and set the device
         # device = 'cuda' if torch.cuda.is_available() else 'cpu'