diff --git a/dataflow/gemma/e2e_test.py b/dataflow/gemma/e2e_test.py
index fdb187d4b9..6586cb48b6 100644
--- a/dataflow/gemma/e2e_test.py
+++ b/dataflow/gemma/e2e_test.py
@@ -42,16 +42,18 @@
 
 from collections.abc import Callable, Iterator
 
+import os
+
 import conftest  # python-docs-samples/dataflow/conftest.py
 from conftest import Utils
 
 import pytest
 
 DATAFLOW_MACHINE_TYPE = "g2-standard-8"
-# TODO(developer): For local testing, ensure the 'gemma_2b_en' directory is uploaded 
-# to a GCS bucket you manage. Update the constant below to point to 
+# TODO(developer): For local testing, ensure the 'gemma_2b_en' directory is uploaded
+# to a GCS bucket you manage. Update the constant below to point to
 # the root path of this uploaded directory (e.g., 'gs://your-bucket-name/path/to/gemma_2b').
-GEMMA_GCS = "gs://perm-dataflow-gemma-example-testdata/gemma_2b"
+GEMMA_GCS = os.getenv("GEMMA_GCS", "gs://perm-dataflow-gemma-example-testdata/gemma_2b")
 NAME = "dataflow/gemma/streaming"
 
 
diff --git a/dataflow/gemma/requirements.txt b/dataflow/gemma/requirements.txt
index 183b6016d7..0a1798ba4d 100644
--- a/dataflow/gemma/requirements.txt
+++ b/dataflow/gemma/requirements.txt
@@ -1,5 +1,4 @@
-protobuf==6.33.6
 apache_beam[gcp]==2.74.0
-keras==3.14.1
+keras==3.15.1
 keras_nlp==0.29.1
-pyOpenSSL==26.3.0
+pyOpenSSL==26.2.0