mazen2100 commited on
Commit
fa64bc4
·
verified ·
1 Parent(s): d77bdee

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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