lenawilli commited on
Commit
44e2127
·
verified ·
1 Parent(s): 810fe68

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -0
src/streamlit_app.py CHANGED
@@ -10,6 +10,10 @@ import pickle
10
  from keras.models import load_model
11
  import os
12
  import requests
 
 
 
 
13
 
14
  from recommendation_utils import (
15
  load_svd_model, load_trainset,
 
10
  from keras.models import load_model
11
  import os
12
  import requests
13
+ import os
14
+
15
+ os.environ["SURPRISE_DATA_FOLDER"] = "/tmp/.surprise_data"
16
+
17
 
18
  from recommendation_utils import (
19
  load_svd_model, load_trainset,