From 26b10f56f33a2dfc50f705247aca2d81e4527cbb Mon Sep 17 00:00:00 2001
From: Stephan Auerhahn <stephan@stability.ai>
Date: Thu, 10 Aug 2023 12:24:12 -0700
Subject: [PATCH] fix missing index

---
 scripts/demo/streamlit_helpers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/demo/streamlit_helpers.py b/scripts/demo/streamlit_helpers.py
index a9ff5e8..119ffd7 100644
--- a/scripts/demo/streamlit_helpers.py
+++ b/scripts/demo/streamlit_helpers.py
@@ -47,6 +47,8 @@ def init_st(spec: SamplingSpec, load_ckpt=True, load_filter=True) -> Dict[str, A
         state["params"] = SamplingParams()
         if load_filter:
             state["filter"] = DeepFloydDataFiltering(verbose=False)
+        else:
+            state["filter"] = None
     return state