Update app.py
Browse files
app.py
CHANGED
@@ -4,10 +4,10 @@ import torch
|
|
4 |
from transformers import pipeline
|
5 |
|
6 |
# Load our Tokenizer
|
7 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
8 |
|
9 |
# Load your model
|
10 |
-
model = "
|
11 |
|
12 |
# load into pipeline
|
13 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
|
|
4 |
from transformers import pipeline
|
5 |
|
6 |
# Load our Tokenizer
|
7 |
+
tokenizer = AutoTokenizer.from_pretrained("nlux/CodeLlama-7b-hf_merge")
|
8 |
|
9 |
# Load your model
|
10 |
+
model = "nlux/CodeLlama-7b-hf_merge"
|
11 |
|
12 |
# load into pipeline
|
13 |
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|