matthoffner commited on
Commit
f1c33ac
Β·
1 Parent(s): e8b2dd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -13,7 +13,7 @@ PromptHelper.compact_text_chunks = compact_text_chunks
13
 
14
  with gr.Blocks(css="") as demo:
15
  with gr.Box():
16
- gr.Markdown("<h1 style='font-size: 48px; text-align: center;'>πŸͺ„ WizardLM PDF Chat πŸ““</h1>")
17
  gr.Markdown("<h3 style='text-align: center;'>πŸ¦™ Local LLM Do It For U πŸ¦™</h3>")
18
 
19
  chat_context = gr.State([])
@@ -55,6 +55,7 @@ with gr.Blocks(css="") as demo:
55
  chat_submit_btn = gr.Button("πŸš€", variant="primary")
56
 
57
  index_select = []
 
58
  chat_input.submit(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
59
  chat_input.submit(reset_textbox, [], [chat_input])
60
  chat_submit_btn.click(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
@@ -69,5 +70,5 @@ with gr.Blocks(css="") as demo:
69
 
70
 
71
  if __name__ == "__main__":
72
- demo.title = "WizardLM Chat"
73
  demo.queue().launch()
 
13
 
14
  with gr.Blocks(css="") as demo:
15
  with gr.Box():
16
+ gr.Markdown("<h1 style='font-size: 48px; text-align: center;'>πŸͺ„ WizardLM Doc Chat πŸ““</h1>")
17
  gr.Markdown("<h3 style='text-align: center;'>πŸ¦™ Local LLM Do It For U πŸ¦™</h3>")
18
 
19
  chat_context = gr.State([])
 
55
  chat_submit_btn = gr.Button("πŸš€", variant="primary")
56
 
57
  index_select = []
58
+ prompt_tmpl = "Default"
59
  chat_input.submit(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
60
  chat_input.submit(reset_textbox, [], [chat_input])
61
  chat_submit_btn.click(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
 
70
 
71
 
72
  if __name__ == "__main__":
73
+ demo.title = "WizardLM Doc Chat"
74
  demo.queue().launch()