Update app.py

app.py

@@ -19,10 +19,11 @@ from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 from diffusers import FluxPipeline
 from typing import Tuple, Dict, Any  # Tuple import 추가
+
 # 파일 상단의 import 문 수정
 import transformers
 from transformers import pipeline as transformers_pipeline
-
+from transformers import Pipeline  # Pipeline 타입 힌트를 위해 추가
 # CUDA 메모리 관리 설정
 torch.cuda.empty_cache()
 torch.backends.cuda.matmul.allow_tf32 = True
@@ -176,7 +177,7 @@ def text_to_image(prompt: str, height: int, width: int, steps: int, scales: floa
     
     with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
         try:
-            generated_image = pipe(
+            generated_image = flux_pipe(  # pipe를 flux_pipe로 변경
                 prompt=[formatted_prompt],
                 generator=torch.Generator().manual_seed(int(seed)),
                 num_inference_steps=int(steps),
@@ -352,7 +353,8 @@ if __name__ == "__main__":
         flux_pipe.fuse_lora(lora_scale=0.125)
         
         # 번역기 초기화
-        translator = transformers_pipeline(  # transformers.pipeline 대신 transformers_pipeline 사용
+        global translator
+        translator = transformers_pipeline(
             "translation", 
             model="Helsinki-NLP/opus-mt-ko-en",
             device=device