Update app.py

app.py

@@ -5,8 +5,8 @@ from StableDiffuser import StableDiffuser
 from tqdm import tqdm
 from train import train
 
-import os
-os.system("nvidia-smi topo -m")
+#import os
+#os.system("nvidia-smi topo -m")
 
 model_map = {'Van Gogh' : 'models/vangogh.pt', 
              'Pablo Picasso': 'models/pablopicasso.pt',
@@ -26,8 +26,8 @@ class Demo:
 
         self.training = False
         
-        print(f'Available GPUs: {torch.cuda.device_count()}')
-        self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda:1').eval().half()
+        #print(f'Available GPUs: {torch.cuda.device_count()}')
+        self.diffuser = StableDiffuser(scheduler='DDIM', seed=42).to('cuda').eval().half()
 
         with gr.Blocks() as demo:
             self.layout()