freddyaboulton HF Staff commited on
Commit
6721334
·
verified ·
1 Parent(s): 2f3a5e7

Update transformers_local.py

Browse files
Files changed (1) hide show
  1. transformers_local.py +3 -0
transformers_local.py CHANGED
@@ -49,12 +49,15 @@ def chat_mistral7b_v0dot3(message: str,
49
  yield "".join(outputs)
50
 
51
 
 
 
52
  # Gradio block
53
  with gr.Blocks() as demo:
54
  gr.Markdown("# ChatInterface with Mistral and Transformers 🤗")
55
  gr.ChatInterface(
56
  fn=chat_mistral7b_v0dot3,
57
  fill_height=True,
 
58
  additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
59
  additional_inputs=[
60
  gr.Slider(minimum=0,
 
49
  yield "".join(outputs)
50
 
51
 
52
+ chatbot = gr.Chatbot(height=400)
53
+
54
  # Gradio block
55
  with gr.Blocks() as demo:
56
  gr.Markdown("# ChatInterface with Mistral and Transformers 🤗")
57
  gr.ChatInterface(
58
  fn=chat_mistral7b_v0dot3,
59
  fill_height=True,
60
+ chatbot=chatbot,
61
  additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
62
  additional_inputs=[
63
  gr.Slider(minimum=0,