Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import subprocess
 from typing import Union
 from huggingface_hub import whoami, HfApi
+from fastapi import FastAPI
+from starlette.middleware.sessions import SessionMiddleware
 
 is_spaces = True if os.environ.get("SPACE_ID") else False
     
@@ -25,11 +27,16 @@ import yaml
 from slugify import slugify
 from transformers import AutoProcessor, AutoModelForCausalLM
 
+# Gradio app 설정
+app = FastAPI()
+app.add_middleware(SessionMiddleware, secret_key="your-secret-key")
+
 if not is_spaces:
     sys.path.insert(0, "ai-toolkit")
     from toolkit.job import get_job
     gr.OAuthProfile = None
     gr.OAuthToken = None
+
     
 MAX_IMAGES = 150
 
@@ -533,4 +540,9 @@ with gr.Blocks(theme=theme, css=css) as demo:
     demo.load(fn=swap_visibilty, outputs=main_ui)
 
 if __name__ == "__main__":
-    demo.launch(share=True, show_error=True)
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        app=app  # FastAPI app 추가
+    )