diff --git a/webui_st.py b/webui_st.py
index 74c426a..eb48c15 100644
--- a/webui_st.py
+++ b/webui_st.py
@@ -10,9 +10,7 @@ from chains.local_doc_qa import LocalDocQA
 from configs.model_config import *
 import nltk
 from models.base import (BaseAnswer,
-                         AnswerResult,
-                         AnswerResultStream,
-                         AnswerResultQueueSentinelTokenListenerQueue)
+                         AnswerResult,)
 import models.shared as shared
 from models.loader.args import parser
 from models.loader import LoaderCheckPoint