Update app.py

app.py

@@ -15,7 +15,27 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 
+# 기존 import문 아래에 추가
+from transformers import pipeline as translation_pipeline
+from diffusers import FluxPipeline
 
+# 초기화 부분에 추가
+def initialize_models():
+    global pipeline, translator, flux_pipe
+    
+    # Trellis 파이프라인 초기화
+    pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
+    pipeline.cuda()
+    
+    # 번역기 초기화
+    translator = translation_pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+    
+    # Flux 파이프라인 초기화
+    flux_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+    flux_pipe.load_lora_weights("gokaygokay/Flux-Game-Assets-LoRA-v2")
+    flux_pipe.fuse_lora(lora_scale=1.0)
+    flux_pipe.to(device="cuda", dtype=torch.bfloat16)
+    
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = "/tmp/Trellis-demo"