diff --git a/action.yml b/action.yml
index 03bc881..0200ea1 100644
--- a/action.yml
+++ b/action.yml
@@ -7,6 +7,9 @@ inputs:
   wandb_api_key:
     description: Your wandb API key available at `/authorize`.  We recommend setting this in github secrets!
     required: true
+  report_template_filename:
+    description: The .py file you want to use for templating.
+    required: true
 runs:
   using: composite
   steps:
@@ -23,7 +26,7 @@ runs:
   - name: Generate a new report
     id: generate-report
     run: |
-      report_url=$(python make_report.py)
+      report_url=$(python ${{ inputs.report_template_filename }}.py)
       echo "REPORT_URL=$report_url" >> $GITHUB_OUTPUT
     shell: bash
     env:
diff --git a/requirements.txt b/requirements.txt
deleted file mode 100644
index b6f1bfa..0000000
--- a/requirements.txt
+++ /dev/null
@@ -1 +0,0 @@
-git+https://github.com/wandb/wandb.git@andrew/report-api-fixes
\ No newline at end of file