ChenyuRabbitLove commited on
Commit
c06edf7
1 Parent(s): 4e14d61

feat/change user model building logic

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,9 +3,12 @@ import gradio as gr
3
  from utils.chatbot import Chatbot, VideoChatbot
4
  from utils.utils import *
5
 
 
 
 
6
  # start of gradio interface
7
  with gr.Blocks() as demo:
8
- user_chatbot = gr.State(Chatbot())
9
  test_video_chabot = gr.State(VideoChatbot())
10
 
11
  with gr.Row():
 
3
  from utils.chatbot import Chatbot, VideoChatbot
4
  from utils.utils import *
5
 
6
+ def create_new_model():
7
+ return Chatbot()
8
+
9
  # start of gradio interface
10
  with gr.Blocks() as demo:
11
+ user_chatbot = gr.State(create_new_model)
12
  test_video_chabot = gr.State(VideoChatbot())
13
 
14
  with gr.Row():