Update app.py

app.py

@@ -18,8 +18,10 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 from diffusers import FluxPipeline
-from transformers import pipeline
 from typing import Tuple, Dict, Any  # Tuple import 추가
+# 파일 상단의 import 문 수정
+import transformers
+from transformers import pipeline as transformers_pipeline
 
 # CUDA 메모리 관리 설정
 torch.cuda.empty_cache()
@@ -328,13 +330,13 @@ if __name__ == "__main__":
 
     try:
         # 3D 생성 파이프라인
-        trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained(  # pipeline을 trellis_pipeline으로 변경
+        trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained(
             "JeffreyXiang/TRELLIS-image-large"
         )
         trellis_pipeline.to(device)
         
         # 이미지 생성 파이프라인
-        flux_pipe = FluxPipeline.from_pretrained(  # pipe를 flux_pipe로 변경
+        flux_pipe = FluxPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.bfloat16,
             device_map="balanced"
@@ -350,7 +352,7 @@ if __name__ == "__main__":
         flux_pipe.fuse_lora(lora_scale=0.125)
         
         # 번역기 초기화
-        translator = transformers.pipeline(  # pipeline을 transformers.pipeline으로 변경
+        translator = transformers_pipeline(  # transformers.pipeline 대신 transformers_pipeline 사용
             "translation", 
             model="Helsinki-NLP/opus-mt-ko-en",
             device=device