checkout tempfile

app.py

@@ -1,10 +1,3 @@
-import os
-# Fix for permission errors in Hugging Face Spaces
-os.environ["STREAMLIT_CONFIG_DIR"] = "/tmp/.streamlit"
-os.environ["STREAMLIT_CACHE_DIR"] = "/tmp/.cache"
-os.makedirs("/tmp/.streamlit", exist_ok=True)
-os.makedirs("/tmp/.cache", exist_ok=True)
-
 import streamlit as st
 import os
 import json
@@ -14,6 +7,9 @@ import torch
 import base64
 from io import BytesIO
 
+import os
+import tempfile
+
 # Configuration (similar to aim.py, but adapted for Streamlit)
 DEFAULT_KEYWORD_COUNT = 5
 DEFAULT_MODEL = "llava-hf/llava-1.5-7b-hf" # A common Llava model on Hugging Face