Update app.py

app.py

@@ -5,6 +5,8 @@ from StableDiffuser import StableDiffuser
 from tqdm import tqdm
 from train import train
 
+!nvidia-smi
+
 model_map = {'Van Gogh' : 'models/vangogh.pt', 
              'Pablo Picasso': 'models/pablopicasso.pt',
              'Car' : 'models/car.pt',
@@ -22,6 +24,7 @@ class Demo:
     def __init__(self) -> None:
 
         self.training = False
+        
         print(f'Available GPUs: {torch.cuda.device_count()}')
         self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:1').eval().half()