phamngoctukts commited on
Commit
190bb2a
·
verified ·
1 Parent(s): 2e18aa4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- tk = "hf" + "_" + "qTOSlDtDtBgJbofv" + "MglsjjhQqbRAYRYnXy"
 
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(