Add nvidia-smi command to app.py for GPU monitoring

app.py

@@ -10,6 +10,7 @@ os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch
 print(torch.__version__)
+os.system("nvidia-smi")
 import numpy as np
 import imageio
 import uuid
@@ -23,8 +24,8 @@ from diffusers import FluxPipeline, AutoencoderKL, AutoencoderTiny
 from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
 
 llm_client = Client("Qwen/Qwen2.5-72B-Instruct")
-device = "cuda" if torch.cuda.is_available() else "cpu"
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
 taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.bfloat16).to(device)
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
 good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)