AC-Angelo93 commited on
Commit
1058a3f
·
verified ·
1 Parent(s): 1116c42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
  # If you have a HF token in the Space secrets, uncomment below:
7
- # os.environ["HUGGINGFACE_HUB_TOKEN"] = os.getenv("HF_TOKEN", "")
8
 
9
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
10
 
 
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
  # If you have a HF token in the Space secrets, uncomment below:
7
+ os.environ["HUGGINGFACE_HUB_TOKEN"] = os.getenv("HF_TOKEN", "")
8
 
9
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
10