fix: updated base

Author: justinmerrell

Dockerfile

@@ -1,12 +1,13 @@
 # Base image -> https://github.com/runpod/containers/blob/main/official-templates/base/Dockerfile
-FROM runpod/base:0.3.1-cuda11.8.0
+# DockerHub -> https://hub.docker.com/r/runpod/base/tags
+FROM runpod/base:0.4.0-cuda11.8.0
 
 # The base image comes with many system dependencies pre-installed to help you get started quickly.
 # Please refer to the base image's Dockerfile for more information before adding additional dependencies.
 # IMPORTANT: The base image overrides the default huggingface cache location.
 
 
-# Optional: System dependencies
+# --- Optional: System dependencies ---
 # COPY builder/setup.sh /setup.sh
 # RUN /bin/bash /setup.sh && \
 #     rm /setup.sh

README.md

@@ -52,6 +52,16 @@ The `CD-docker_dev.yml` file will build the image and push it to the `dev` tag,
 
 The `CI-test_worker.yml` file will test the worker using the input provided by the `--test_input` argument when calling the file containing your handler. Be sure to update this workflow to install any dependencies you need to run your tests.
 
+## Example Input
+
+```json
+{
+    "input": {
+        "name": "John Doe"
+    }
+}
+```
+
 ## 💡 | Best Practices
 
 System dependency installation, model caching, and other shell tasks should be added to the `builder/setup.sh` this will allow you to easily setup your Dockerfile as well as run CI/CD tasks.