zhaozitian commited on
Commit
0e4a922
·
1 Parent(s): 02dc13e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -3,6 +3,9 @@ from peft import PeftModel
3
  import transformers
4
  import gradio as gr
5
 
 
 
 
6
  assert (
7
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
8
  ), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"
 
3
  import transformers
4
  import gradio as gr
5
 
6
+ from huggingface_hub import notebook_login
7
+ notebook_login()
8
+
9
  assert (
10
  "LlamaTokenizer" in transformers._import_structure["models.llama"]
11
  ), "LLaMA is now in HuggingFace's main branch.\nPlease reinstall it: pip uninstall transformers && pip install git+https://github.com/huggingface/transformers.git"