Chittrarasu commited on
Commit
7a96f21
·
1 Parent(s): d834c46
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -4,6 +4,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
  import torch
5
  import matplotlib.pyplot as plt
6
  import time
 
 
7
 
8
  # Load model and tokenizer
9
  model_name = "tabularisai/multilingual-sentiment-analysis"
 
4
  import torch
5
  import matplotlib.pyplot as plt
6
  import time
7
+ import os
8
+ os.environ["HF_HOME"] = "/app/hf_cache"
9
 
10
  # Load model and tokenizer
11
  model_name = "tabularisai/multilingual-sentiment-analysis"