Update app.py

app.py

@@ -11,11 +11,14 @@ import gradio as gr
 import torch
 from diffusers import FluxPipeline
 from PIL import Image
+# Setup and initialization code
+cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
+# 이 부분을 수정
 
 # Setup and initialization code
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
 # Change gallery path to user's home directory for persistence
-gallery_path = path.join(os.path.expanduser("~"), "ai_generated_images")
+gallery_path = path.join(os.getcwd(), "gallery")  # 현재 작업 디렉토리 내의 gallery 폴더로 변경
 os.environ["TRANSFORMERS_CACHE"] = cache_path
 os.environ["HF_HUB_CACHE"] = cache_path
 os.environ["HF_HOME"] = cache_path
@@ -358,4 +361,4 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(allowed_paths=[gallery_path])  # allowed_paths 파라미터 추가