Spaces:
Sleeping
Sleeping
kovacsvi
commited on
Commit
·
abf66d1
1
Parent(s):
fc96b01
don't use utils.py yet
Browse files
app.py
CHANGED
|
@@ -13,7 +13,6 @@ import matplotlib.colors as mcolors
|
|
| 13 |
import plotly.express as px
|
| 14 |
import seaborn as sns
|
| 15 |
from tqdm import tqdm
|
| 16 |
-
from utils import css
|
| 17 |
|
| 18 |
PATH = '/data/' # at least 150GB storage needs to be attached
|
| 19 |
os.environ['TRANSFORMERS_CACHE'] = PATH
|
|
@@ -21,6 +20,10 @@ os.environ['HF_HOME'] = PATH
|
|
| 21 |
os.environ['HF_DATASETS_CACHE'] = PATH
|
| 22 |
os.environ['TORCH_HOME'] = PATH
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
HF_TOKEN = os.environ["hf_read"]
|
| 25 |
|
| 26 |
SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment", 2: "Positive"}
|
|
|
|
| 13 |
import plotly.express as px
|
| 14 |
import seaborn as sns
|
| 15 |
from tqdm import tqdm
|
|
|
|
| 16 |
|
| 17 |
PATH = '/data/' # at least 150GB storage needs to be attached
|
| 18 |
os.environ['TRANSFORMERS_CACHE'] = PATH
|
|
|
|
| 20 |
os.environ['HF_DATASETS_CACHE'] = PATH
|
| 21 |
os.environ['TORCH_HOME'] = PATH
|
| 22 |
|
| 23 |
+
css = '''
|
| 24 |
+
.info {font-size: 3em; !important}
|
| 25 |
+
'''
|
| 26 |
+
|
| 27 |
HF_TOKEN = os.environ["hf_read"]
|
| 28 |
|
| 29 |
SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment", 2: "Positive"}
|
utils.py
CHANGED
|
@@ -27,8 +27,4 @@ def download_hf_models():
|
|
| 27 |
token=HF_TOKEN)
|
| 28 |
for tokenizer_id in tokenizers:
|
| 29 |
AutoTokenizer.from_pretrained(tokenizer_id)
|
| 30 |
-
|
| 31 |
-
css = '''
|
| 32 |
-
.info {font-size: 3em; !important}
|
| 33 |
-
'''
|
| 34 |
|
|
|
|
| 27 |
token=HF_TOKEN)
|
| 28 |
for tokenizer_id in tokenizers:
|
| 29 |
AutoTokenizer.from_pretrained(tokenizer_id)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|