fixed for hf space
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
|
2 |
import streamlit as st
|
3 |
import os
|
4 |
-
os.environ['HF_HOME'] = '/scratch/sroydip1/cache/hf/'
|
5 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets["HF_TOKEN"]
|
6 |
# import torch
|
7 |
import pickle
|
|
|
1 |
|
2 |
import streamlit as st
|
3 |
import os
|
4 |
+
# os.environ['HF_HOME'] = '/scratch/sroydip1/cache/hf/'
|
5 |
os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets["HF_TOKEN"]
|
6 |
# import torch
|
7 |
import pickle
|