Spaces:
Sleeping
Sleeping
Allen Park
commited on
Commit
·
0703e71
1
Parent(s):
64a90e6
add HF_TOKEN env variable
Browse files
app.py
CHANGED
|
@@ -1,8 +1,10 @@
|
|
|
|
|
| 1 |
import torch
|
| 2 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 3 |
import gradio as gr
|
| 4 |
import spaces
|
| 5 |
|
|
|
|
| 6 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 7 |
|
| 8 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-Nemo-Instruct-2407")
|
|
|
|
| 1 |
+
import os
|
| 2 |
import torch
|
| 3 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 4 |
import gradio as gr
|
| 5 |
import spaces
|
| 6 |
|
| 7 |
+
HF_TOKEN = os.environ.get("HF_TOKEN", None)
|
| 8 |
device = "cuda" # for GPU usage or "cpu" for CPU usage
|
| 9 |
|
| 10 |
tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-Nemo-Instruct-2407")
|