diff --git a/lit_nlp/examples/gcp/model_server_test.py b/lit_nlp/examples/gcp/model_server_test.py
index f0e1291b..bd9ee6cb 100644
--- a/lit_nlp/examples/gcp/model_server_test.py
+++ b/lit_nlp/examples/gcp/model_server_test.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 
+import io
 import os
 from unittest import mock
 
@@ -30,6 +31,7 @@ class TestWSGIApp(parameterized.TestCase):
   def setUpClass(cls):
     super().setUpClass()
 
+    _ = io.StringIO('temp data')
     test_model_name = 'lit_on_gcp_test_model'
     sal_name, tok_name = pd_utils.generate_model_group_names(test_model_name)
     test_model_config = f'{test_model_name}:test_model_path'
diff --git a/pyproject.toml b/pyproject.toml
index 66536476..6db282c6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -41,6 +41,10 @@ dependencies = [
   "termcolor>=2.3.0",
   "tqdm>=4.64.0",
   "werkzeug>=2.2.3",
+  "decorator>=4.4.0",
+  "contextlib2>=21.6.0",
+  "fsspec>=2023.9.0",
+  "colorama>=0.4.4",  
 ]
 # LINT.ThenChange(./requirements.txt)
 classifiers = [
diff --git a/requirements.txt b/requirements.txt
index caabea60..d35ede61 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -38,4 +38,8 @@ six>=1.16.0
 termcolor>=2.3.0
 tqdm>=4.64.0
 werkzeug>=2.2.3
+decorator>=4.4.0,
+contextlib2>=21.6.0,
+fsspec>=2023.9.0,
+colorama>=0.4.4,  
 # LINT.ThenChange(./pyproject.toml)