Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import gradio.helpers
 from datasets import load_dataset
 
 import base64
@@ -15,6 +16,8 @@ from share_btn import community_icon_html, loading_icon_html, share_js
 word_list_dataset = load_dataset("google/word-list-sd", data_files="list.txt", use_auth_token=True)
 word_list = word_list_dataset["train"]['text']
 
+gradio.helpers.CACHE_FOLDER="/data/cache"
+
 def infer(prompt, negative="low_quality", scale=7, profile: gr.OAuthProfile | None = None):
     for filter in word_list:
         if re.search(rf"\b{filter}\b", prompt):