Update app.py

app.py

@@ -318,13 +318,14 @@ if __name__ == "__main__":
         "JeffreyXiang/TRELLIS-image-large",
         use_auth_token=HF_TOKEN
     )
-    pipeline.cuda()
+    pipeline.to("cuda")  # .cuda() 대신 .to("cuda") 사용
     
     # 이미지 생성 파이프라인
     pipe = FluxPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-dev",
         torch_dtype=torch.bfloat16,
-        use_auth_token=HF_TOKEN
+        use_auth_token=HF_TOKEN,
+        device_map="auto"  # device_map 추가
     )
     
     # Hyper-SD LoRA 로드
@@ -336,7 +337,11 @@ if __name__ == "__main__":
         )
     )
     pipe.fuse_lora(lora_scale=0.125)
-    pipe.to(device="cuda", dtype=torch.bfloat16)
+    
+    # 번역기 초기화 (GPU 설정 추가)
+    translator = pipeline("translation", 
+                        model="Helsinki-NLP/opus-mt-ko-en",
+                        device=0 if torch.cuda.is_available() else -1)
     
     try:
         pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))