Update app.py

app.py

@@ -4,6 +4,8 @@ from datasets import load_dataset
 from PIL import Image
 import numpy as np
 
+
+        
 # 初始化模型 (缓存)
 @st.cache_resource
 def load_models():
@@ -20,6 +22,15 @@ def load_style_examples():
     dataset = load_dataset("AntZet/home_decoration_objects_images")
     return dataset['train'].to_pandas()
 
+@st.cache_data
+def load_style_examples():
+    try:
+        dataset = load_dataset("AntZet/home_decoration_objects_images", streaming=True)
+        return dataset['train'].to_pandas()
+    except:
+        return pd.DataFrame(columns=['style', 'image'])  # 返回空数据集作为fallback
+
+        
 # 主函数
 def main():
     st.title("🏠 AI 装修风格匹配器")