Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,12 @@
 import streamlit as st
 from diffusers import StableDiffusionPipeline
 import torch
+import os
+
+# Set cache to writable directory
+os.environ["HF_HOME"] = "/tmp"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+os.environ["HF_HUB_CACHE"] = "/tmp"
 
 st.set_page_config(page_title="AI Design Generator", layout="centered")
 
@@ -10,7 +16,7 @@ def load_model():
         "runwayml/stable-diffusion-v1-5", 
         torch_dtype=torch.float32
     )
-    model.to("cpu")  # Space is on CPU unless you enable GPU manually
+    model.to("cpu")  # CPU unless GPU enabled in Space settings
     return model
 
 pipe = load_model()
@@ -22,5 +28,3 @@ if st.button("Generate Image"):
     with st.spinner("Generating..."):
         image = pipe(prompt).images[0]
         st.image(image, caption="Generated Design", use_column_width=True)
-
-