changes in app.py
Browse files
app.py
CHANGED
|
@@ -20,6 +20,7 @@ pipeline = transformers.pipeline(
|
|
| 20 |
torch_dtype=torch.float16,
|
| 21 |
device_map="auto",
|
| 22 |
do_sample=True,
|
|
|
|
| 23 |
)
|
| 24 |
|
| 25 |
def get_llama_response(prompt):
|
|
|
|
| 20 |
torch_dtype=torch.float16,
|
| 21 |
device_map="auto",
|
| 22 |
do_sample=True,
|
| 23 |
+
token=HF_TOKEN,
|
| 24 |
)
|
| 25 |
|
| 26 |
def get_llama_response(prompt):
|