Spaces:
Runtime error
Runtime error
phamngoctukts
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,8 @@ from transformers import pipeline
|
|
14 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
|
15 |
import torch
|
16 |
from huggingface_hub import login
|
17 |
-
|
|
|
18 |
login(tk)
|
19 |
model_id = "meta-llama/Llama-3.2-1B"
|
20 |
text2text = pipeline(
|
|
|
14 |
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModel
|
15 |
import torch
|
16 |
from huggingface_hub import login
|
17 |
+
import os
|
18 |
+
tk = token = os.environ.get("HF_TOKEN")
|
19 |
login(tk)
|
20 |
model_id = "meta-llama/Llama-3.2-1B"
|
21 |
text2text = pipeline(
|