mikeee commited on
Commit
556ee99
1 Parent(s): 0ad820c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import os
2
  os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
3
 
4
-
5
  import gradio as gr
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
 
@@ -32,5 +31,4 @@ print(f"The model '{model_name}' is taking approximately {model_size_gb:.2f} GB
32
 
33
  # msg.submit(respond, [msg, chatbot], [msg, chatbot])
34
 
35
- # if __name__ == "__main__":
36
- # demo.launch()
 
1
  import os
2
  os.system("pip install --upgrade torch transformers sentencepiece scipy cpm_kernels accelerate bitsandbytes loguru")
3
 
 
4
  import gradio as gr
5
  from transformers import AutoTokenizer, AutoModelForCausalLM
6
 
 
31
 
32
  # msg.submit(respond, [msg, chatbot], [msg, chatbot])
33
 
34
+ # demo.launch()