Update app.py

app.py

@@ -38,7 +38,6 @@ def initialize_models(device):
     g.trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained(
         "JeffreyXiang/TRELLIS-image-large"
     )
-    g.trellis_pipeline.to(device)
     
     # 이미지 생성 파이프라인
     g.flux_pipe = FluxPipeline.from_pretrained(
@@ -69,7 +68,6 @@ torch.backends.cuda.matmul.allow_tf32 = True
 torch.backends.cudnn.benchmark = True
 
 # 환경 변수 설정
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:512"
 os.environ['SPCONV_ALGO'] = 'native'
 os.environ['SPARSE_BACKEND'] = 'native'
@@ -422,19 +420,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
-    # CUDA 사용 가능 여부 확인
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    print(f"Using device: {device}")
-
     try:
+        # CPU 모드로 초기화
+        device = "cpu"
+        print(f"Using device: {device}")
+        
         # 모델 초기화
         initialize_models(device)
         
-        # CUDA 메모리 초기화
-        if torch.cuda.is_available():
-            torch.cuda.empty_cache()
-            torch.cuda.synchronize()
-        
         # 초기 이미지 전처리 테스트
         try:
             test_image = Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))