Update app.py

app.py

@@ -328,17 +328,16 @@ if __name__ == "__main__":
 
     try:
         # 3D 생성 파이프라인
-        pipeline = TrellisImageTo3DPipeline.from_pretrained(
+        trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained(  # pipeline을 trellis_pipeline으로 변경
             "JeffreyXiang/TRELLIS-image-large"
         )
-        pipeline.to(device)
+        trellis_pipeline.to(device)
         
         # 이미지 생성 파이프라인
-        pipe = FluxPipeline.from_pretrained(
+        flux_pipe = FluxPipeline.from_pretrained(  # pipe를 flux_pipe로 변경
             "black-forest-labs/FLUX.1-dev",
             torch_dtype=torch.bfloat16,
-            use_auth_token=HF_TOKEN,
-            device_map="balanced"  # "auto"에서 "balanced"로 변경
+            device_map="balanced"
         )
         
         # Hyper-SD LoRA 로드
@@ -347,11 +346,11 @@ if __name__ == "__main__":
             "Hyper-FLUX.1-dev-8steps-lora.safetensors",
             use_auth_token=HF_TOKEN
         )
-        pipe.load_lora_weights(lora_path)
-        pipe.fuse_lora(lora_scale=0.125)
+        flux_pipe.load_lora_weights(lora_path)
+        flux_pipe.fuse_lora(lora_scale=0.125)
         
         # 번역기 초기화
-        translator = pipeline(
+        translator = transformers.pipeline(  # pipeline을 transformers.pipeline으로 변경
             "translation", 
             model="Helsinki-NLP/opus-mt-ko-en",
             device=device
@@ -364,7 +363,7 @@ if __name__ == "__main__":
         # 초기 이미지 전처리 테스트
         try:
             test_image = Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8))
-            pipeline.preprocess_image(test_image)
+            trellis_pipeline.preprocess_image(test_image)
         except Exception as e:
             print(f"Warning: Initial preprocessing test failed: {e}")