Update app.py

app.py

@@ -1,3 +1,5 @@
+
+
 import os
 import gradio as gr
 import random
@@ -10,21 +12,21 @@ import numpy as np
 from diffusers import DiffusionPipeline
 from transformers import pipeline as hf_pipeline
 
-# ---------------------- ZeroGPU 환경에서 GPU 사용 설정 ----------------------
-# Hugging Face Spaces의 ZeroGPU 환경이면 환경 변수 ZERO_GPU가 설정되어 있다고 가정
+# ---------------------- ZeroGPU 환경에서 CPU 사용 설정 ----------------------
 if os.getenv("ZERO_GPU"):
-    device = "cuda"
-    torch.cuda.set_device(0)
+    device = "cpu"
 else:
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-dtype = torch.bfloat16 if device == "cuda" else torch.float32
+dtype = torch.bfloat16 if (device == "cuda" and torch.cuda.is_available()) else torch.float32
 
 # ---------------------- 이미지 생성 관련 설정 ----------------------
-# 한국어-영어 번역 모델 로드 (장치에 따라 CPU 또는 GPU 사용)
-translator = hf_pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=0 if device=="cuda" else -1)
+translator = hf_pipeline(
+    "translation", 
+    model="Helsinki-NLP/opus-mt-ko-en",
+    device=0 if device == "cuda" else -1
+)
 
-# Diffusion Pipeline 로드 (예: FLUX.1-schnell 모델)
 pipe = DiffusionPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-schnell",
     torch_dtype=dtype
@@ -71,7 +73,6 @@ def generate_design_image(prompt, seed=42, randomize_seed=True, width=1024, heig
     return image
 
 # ---------------------- Gemini API 및 아이디어 생성 관련 기존 코드 ----------------------
-# 로깅 설정
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
@@ -82,7 +83,6 @@ logging.basicConfig(
 )
 logger = logging.getLogger("idea_generator")
 
-# Gemini API 키는 환경 변수 GEMINI_API_KEY에서 가져옵니다.
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)
 
@@ -107,6 +107,8 @@ def choose_alternative(transformation):
     else:
         return random.choice([left, right])
 
+# 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (생략된 부분 동일)...
+
 # 창의적인 모델/컨셉/형상 변화 아이디어를 위한 카테고리 (총 15개)
 physical_transformation_categories = {
     "공간 이동": [
@@ -249,6 +251,7 @@ physical_transformation_categories = {
     ]
 }
 
+
 ##############################################################################
 # Gemini API 호출 함수 (모델: gemini-2.0-flash-thinking-exp-01-21)
 ##############################################################################