research14 commited on
Commit
3e98569
·
1 Parent(s): 61a4fb2

more chatbots

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -10,9 +10,13 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
10
 
11
  with gr.Blocks() as demo:
12
  with gr.Row():
13
- vicuna_chatbot = gr.Chatbot(label="Vicuna", live=True)
14
- msg = gr.Textbox()
15
- clear = gr.ClearButton([msg, vicuna_chatbot])
 
 
 
 
16
 
17
  def respond(message, chat_history, chatbot_idx):
18
  input_ids = tokenizer.encode(message, return_tensors="pt")
@@ -22,6 +26,6 @@ with gr.Blocks() as demo:
22
  time.sleep(2)
23
  return "", chat_history
24
 
25
- msg.submit(respond, [msg, vicuna_chatbot, vicuna_chatbot])
26
 
27
  demo.launch()
 
10
 
11
  with gr.Blocks() as demo:
12
  with gr.Row():
13
+ vicuna_chatbot = gr.Chatbot(label="vicuna-7b", live=False)
14
+ llama_chatbot = gr.Chatbot(label="llama-7b", live=False)
15
+ gpt_chatbot = gr.Chatbot(label="gpt-3.5", live=False)
16
+ with gr.Row():
17
+ prompt = gr.Textbox(show_label=False, placeholder="Enter prompt")
18
+ send_button_Chunk = gr.Button("Send", scale=0)
19
+ clear = gr.ClearButton([prompt, vicuna_chatbot])
20
 
21
  def respond(message, chat_history, chatbot_idx):
22
  input_ids = tokenizer.encode(message, return_tensors="pt")
 
26
  time.sleep(2)
27
  return "", chat_history
28
 
29
+ prompt.submit(respond, [prompt, vicuna_chatbot, vicuna_chatbot])
30
 
31
  demo.launch()