Update app.py

app.py

@@ -234,7 +234,6 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
                 # 파이프라인을 GPU로 이동
                 g.trellis_pipeline.to('cuda')
                 
-                # batch_size 제거하고 파라미터 수정
                 outputs = g.trellis_pipeline.run(
                     image,
                     seed=seed,
@@ -266,9 +265,12 @@ def image_to_3d(trial_id: str, seed: int, randomize_seed: bool, ss_guidance_stre
                     resolution=384
                 )['normal']
                 
-                # CPU로 데이터 이동 및 메모리 정리
-                video = [v.cpu().numpy() for v in video]
-                video_geo = [v.cpu().numpy() for v in video_geo]
+                # tensor를 numpy로 변환
+                if torch.is_tensor(video[0]):
+                    video = [v.cpu().numpy() if torch.is_tensor(v) else v for v in video]
+                if torch.is_tensor(video_geo[0]):
+                    video_geo = [v.cpu().numpy() if torch.is_tensor(v) else v for v in video_geo]
+                
                 clear_gpu_memory()
                 
                 # 비디오 생성 및 저장