Update app.py

app.py

@@ -1,6 +1,5 @@
 import subprocess
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
-subprocess.run('nvidia-smi',shell=True)
 
 import gradio as gr
 import spaces
@@ -8,8 +7,11 @@ from gradio_litmodel3d import LitModel3D
 
 import os
 os.environ['SPCONV_ALGO'] = 'native'
+os.system("nvidia-smi")
+os.system("nvcc --version")
 from typing import *
 import torch
+print(f"Torch version {torch.__version__}")
 import numpy as np
 import imageio
 import uuid