themehmi commited on
Commit
613f1c3
·
verified ·
1 Parent(s): e13b96a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,6 +3,9 @@ import torch
3
  import os
4
  import shutil
5
  import subprocess
 
 
 
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  from langchain_text_splitters import RecursiveCharacterTextSplitter, Language
8
  from langchain_core.documents import Document
@@ -288,7 +291,7 @@ with gr.Blocks(title="Codebase Assistant") as demo:
288
  clear = gr.Button("🗑️ Clear Chat", scale=1)
289
 
290
  msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
291
- clear.click(lambda: None, None, chatbot, queue=False)
292
 
293
  if __name__ == "__main__":
294
  demo.launch(server_name="0.0.0.0", css=custom_css, theme=gr.themes.Soft())
 
3
  import os
4
  import shutil
5
  import subprocess
6
+
7
+ os.environ["TOKENIZERS_PARALLELISM"] = "false"
8
+
9
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
10
  from langchain_text_splitters import RecursiveCharacterTextSplitter, Language
11
  from langchain_core.documents import Document
 
291
  clear = gr.Button("🗑️ Clear Chat", scale=1)
292
 
293
  msg.submit(respond, inputs=[msg, chatbot], outputs=[msg, chatbot])
294
+ clear.click(lambda: ("", []), inputs=None, outputs=[msg, chatbot], queue=False)
295
 
296
  if __name__ == "__main__":
297
  demo.launch(server_name="0.0.0.0", css=custom_css, theme=gr.themes.Soft())