Update app.py

app.py

@@ -14,7 +14,6 @@ from contextlib import contextmanager
 from gradio.helpers import create_examples 
 import subprocess
 
-import torch; print(torch.cuda.device_count()) 
 
 from pathlib import Path
 import os
@@ -199,6 +198,8 @@ from typing import Optional
 from pathlib import Path
 import torchaudio
 
+import torch; print(torch.cuda.device_count()) 
+
 from huggingface_hub import hf_hub_download, snapshot_download
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_core.model.video_vae import TilingConfig