Update src/streamlit_app.py
Browse files- src/streamlit_app.py +8 -0
src/streamlit_app.py
CHANGED
@@ -1,3 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
import streamlit as st
|
2 |
import torch
|
3 |
import numpy as np
|
|
|
1 |
+
import os
|
2 |
+
|
3 |
+
# Fix for permissions on Hugging Face Spaces
|
4 |
+
os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf'
|
5 |
+
os.environ['HF_HOME'] = '/tmp/hf'
|
6 |
+
os.environ['XDG_CACHE_HOME'] = '/tmp'
|
7 |
+
os.environ['STREAMLIT_HOME'] = '/tmp'
|
8 |
+
os.makedirs('/tmp/hf', exist_ok=True)
|
9 |
import streamlit as st
|
10 |
import torch
|
11 |
import numpy as np
|