Update app.py

app.py

@@ -37,6 +37,8 @@ flux_pipe.vae.enable_slicing()
 flux_pipe.vae.enable_tiling()
 flux_pipe.to(torch.float16)
 
+
+@spaces.GPU(duration=120)
 def generate_image(prompt, guidance_scale, width, height):
     # 이미지를 생성하는 함수
     output_image = flux_pipe(
@@ -49,7 +51,7 @@ def generate_image(prompt, guidance_scale, width, height):
     ).images[0]
 
     # 결과 폴더 생성
-    result_folder = "result_flux"
+    result_folder = "/tmp/flux/"
     os.makedirs(result_folder, exist_ok=True)
 
     # 파일 이름 생성
@@ -58,8 +60,8 @@ def generate_image(prompt, guidance_scale, width, height):
     filename = f"{'_'.join(prompt.split()[:3])}_{timestamp}.png"
     output_path = os.path.join(result_folder, filename)
 
-    # 이미지를 저장
-    output_image.save(output_path)
+    # # 이미지를 저장
+    # output_image.save(output_path)
 
     return output_image, output_path  # 두 개의 출력 반환