diff --git a/frontend/chat_page.py b/frontend/chat_page.py index dfe3e17..ebd8bbf 100644 --- a/frontend/chat_page.py +++ b/frontend/chat_page.py @@ -1,4 +1,7 @@ import gradio as gr +import sys +from pathlib import Path +sys.path.append(str(Path(__file__).resolve().parent.parent)) from global_var import model,tokenizer def chat_page(): @@ -23,6 +26,10 @@ def chat_page(): msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then( bot, chatbot, chatbot + ) clear.click(lambda: None, None, chatbot, queue=False) - return demo \ No newline at end of file + return demo + +if __name__ == "__main__": + chat_page().queue().launch() \ No newline at end of file diff --git a/frontend/prompt_manage_page.py b/frontend/prompt_manage_page.py index e52576d..a2426c7 100644 --- a/frontend/prompt_manage_page.py +++ b/frontend/prompt_manage_page.py @@ -1,5 +1,8 @@ import gradio as gr +import sys +from pathlib import Path from typing import List +sys.path.append(str(Path(__file__).resolve().parent.parent)) from global_var import prompt_store from schema.prompt import promptTempleta def prompt_manage_page(): @@ -113,4 +116,10 @@ def prompt_manage_page(): outputs=[prompt_table] ) - return demo \ No newline at end of file + return demo + + +if __name__ == "__main__": + demo = prompt_manage_page() + demo.queue() + demo.launch() \ No newline at end of file