Update app.py

app.py

@@ -7,6 +7,7 @@ subprocess.run(
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True,
 )
+subprocess.run("nvidia-smi")
 from transformers import AutoTokenizer, AutoModel
 import torch
 import torchvision.transforms as T