alamshoaib134 commited on
Commit
41c74da
·
verified ·
1 Parent(s): 2088fae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -10,6 +10,10 @@ import transformers
10
  import torch
11
  import textwrap
12
 
 
 
 
 
13
  model = "meta-llama/Llama-2-7b-chat-hf"
14
  tokenizer = AutoTokenizer.from_pretrained(model)
15
  pipeline = transformers.pipeline(
 
10
  import torch
11
  import textwrap
12
 
13
+ from huggingface_hub import login
14
+ login(token = HF_TOKEN)
15
+
16
+
17
  model = "meta-llama/Llama-2-7b-chat-hf"
18
  tokenizer = AutoTokenizer.from_pretrained(model)
19
  pipeline = transformers.pipeline(