diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
index f5355bc..ef146f3 100644
--- a/scripts/demo/sampling.py
+++ b/scripts/demo/sampling.py
@@ -1,7 +1,13 @@
-from dataclasses import asdict
+import os
+
+import numpy as np
+import streamlit as st
+import torch
+from einops import repeat
 from pytorch_lightning import seed_everything
 
 from sgm.inference.api import (
+    SamplingSpec,
     SamplingParams,
     ModelArchitecture,
     SamplingPipeline,
@@ -11,7 +17,17 @@ from sgm.inference.helpers import (
     get_unique_embedder_keys_from_conditioner,
     perform_save_locally,
 )
-from scripts.demo.streamlit_helpers import *
+from scripts.demo.streamlit_helpers import (
+    get_interactive_image,
+    get_unique_embedder_keys_from_conditioner,
+    init_embedder_options,
+    init_sampling,
+    init_save_locally,
+    init_st,
+    perform_save_locally,
+    set_lowvram_mode,
+    show_samples,
+)
 
 SAVE_PATH = "outputs/demo/txt2img/"