ruchi commited on
Commit
9ff0d9c
·
1 Parent(s): da5916b

Fix error bracket

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
7
  tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
8
  model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
9
 
10
- pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=200))
11
 
12
 
13
  # Create a banner using Markdown
 
7
  tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
8
  model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
9
 
10
+ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_length=200)
11
 
12
 
13
  # Create a banner using Markdown