Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from einops import rearrange, repeat
 from PIL import Image
 import numpy as np
+import spaces  # Hugging Face Spaces 임포트 추가
 
 from flux.sampling import denoise, get_noise, get_schedule, prepare, rf_denoise, rf_inversion, unpack
 from flux.util import (
@@ -71,6 +72,8 @@ except Exception as e:
     model_initialized = False
 
 
+# Spaces GPU 데코레이터 추가 (120초 GPU 사용)
+@spaces.GPU(duration=120)
 @torch.inference_mode()
 def generate_image(
     prompt: str,
@@ -247,7 +250,7 @@ def generate_image(
 
 
 def create_demo():
-    with gr.Blocks(theme="apriel") as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("# PuLID: 인물 이미지 변환 도구")
         
         if not model_initialized:
@@ -278,7 +281,28 @@ def create_demo():
                 seed_output = gr.Textbox(label="결과/오류 메시지")
                 gr.Markdown(_CITE_)
 
-        # Gradio 이벤트 연결 수정 - 직접 정수 값을 전달하지 않고 UI 컴포넌트만 전달
+        # 예제 추가
+        with gr.Row():
+            gr.Markdown("## 예제")
+            example_inps = [
+                [
+                    'a portrait of a clown',
+                    'example_inputs/unsplash/lhon-karwan-11tbHtK5STE-unsplash.jpg',
+                    0.5, 3.5, "-1", 16, 0.4, "", 3.5, 0.5, 0.8
+                ],
+                [
+                    'a portrait of a zombie',
+                    'example_inputs/unsplash/baruk-granda-cfLL_jHQ-Iw-unsplash.jpg',
+                    0.4, 3.5, "42", 16, 0.4, "", 3.5, 0.5, 0.8
+                ]
+            ]
+            gr.Examples(
+                examples=example_inps,
+                inputs=[prompt, id_image, id_weight, guidance, seed, num_steps, 
+                        id_weight, neg_prompt, true_cfg, gamma, eta]
+            )
+
+        # Gradio 이벤트 연결
         generate_btn.click(
             fn=generate_image,
             inputs=[
@@ -300,5 +324,7 @@ if __name__ == "__main__":
     parser.add_argument("--port", type=int, default=8080)
     args = parser.parse_args()
 
+    print("Hugging Face Spaces 환경에서 실행 중입니다. GPU 할당을 요청합니다.")
+    
     demo = create_demo()
-    demo.launch(ssr_mode=False)
+    demo.queue(concurrency_count=1).launch(server_name="0.0.0.0", server_port=args.port)