Upload app.py

app.py

@@ -3,16 +3,17 @@ import gradio as gr
 import torch
 from diffusers import FluxPipeline
 
-# 모델 로드 (Space 시작할 때 한 번만)
+# CPU에 로드 (ZeroGPU는 GPU를 함수 실행할 때만 붙여줌)
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.bfloat16,
 )
 pipe.load_lora_weights("Cpldxx/onesomestyle-dora")
-pipe.to("cuda")
+# pipe.to("cuda") ← 이거 지워야 함
 
 @spaces.GPU(duration=120)
 def generate(prompt, steps=28, guidance=3.5, width=1024, height=1024, seed=-1):
+    pipe.to("cuda")  # ← 여기서 GPU로 이동
     if seed == -1:
         seed = torch.randint(0, 2**32, (1,)).item()
     generator = torch.Generator("cuda").manual_seed(seed)
@@ -45,4 +46,3 @@ with gr.Blocks() as demo:
     btn.click(generate, inputs=[prompt, steps, guidance, width, height, seed], outputs=output)
 
 demo.launch()
-