Update app.py

app.py

@@ -4,11 +4,13 @@ import PIL.Image
 import numpy as np
 import random
 import torch
+import subprocess
 
 ldm_pipeline = LDMTextToImagePipeline.from_pretrained("CompVis/ldm-text2im-large-256")
 
 def predict(prompt, steps=100, seed=42, guidance_scale=6.0):
     torch.cuda.empty_cache()
+    print(subprocess.check_output(["nvidia-smi"], stderr=subprocess.STDOUT).decode("utf8"))
     generator = torch.manual_seed(seed)
     images = ldm_pipeline([prompt], generator=generator, num_inference_steps=steps, eta=0.3, guidance_scale=guidance_scale)["sample"]
     return images[0]