Update app.py

app.py

@@ -10,6 +10,9 @@ import numpy as np
 from diffusers import DiffusionPipeline
 from transformers import pipeline as hf_pipeline
 
+##############################################################################
+# 1) ZeroGPU 환경 처리 + device, dtype 설정
+##############################################################################
 ##############################################################################
 # 1) ZeroGPU 환경 처리 + device, dtype 설정
 ##############################################################################
@@ -17,17 +20,19 @@ from transformers import pipeline as hf_pipeline
 try:
     import zerogpu
     zerogpu.init()
+    print("ZeroGPU initialized successfully")
     device = "cuda" if torch.cuda.is_available() else "cpu"
 except ImportError:
     # ZeroGPU가 설치되지 않은 경우
+    print("ZeroGPU package not installed, continuing without it")
     if os.getenv("ZERO_GPU"):
         print("ZeroGPU environment variable is set but zerogpu package is not installed.")
-        device = "cpu"
-    else:
-        device = "cuda" if torch.cuda.is_available() else "cpu"
+    device = "cuda" if torch.cuda.is_available() else "cpu"
 except Exception as e:
+    # ZeroGPU 초기화 중 다른 오류가 발생한 경우
     print(f"Error initializing ZeroGPU: {e}")
-    device = "cpu"
+    print("Continuing without ZeroGPU")
+    device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # GPU일 때만 bfloat16, 그 외에는 float32
 dtype = torch.bfloat16 if device == "cuda" else torch.float32