Spaces:
Sleeping
Sleeping
ruchi
commited on
Commit
·
9ff0d9c
1
Parent(s):
da5916b
Fix error bracket
Browse files
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
|