Hugging Face's logo

Spaces:
baulab
/
Erasing-Concepts-In-Diffusion
Runtime error

App Files Community
5
RohitGandikota commited on
Commit
079d713
·
1 Parent(s): e0bcdbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from tqdm import tqdm
6
  from train import train
7
 
8
  import os
9
- os.system("nvidia-smi")
10
 
11
  model_map = {'Van Gogh' : 'models/vangogh.pt',
12
  'Pablo Picasso': 'models/pablopicasso.pt',
 
6
  from train import train
7
 
8
  import os
9
+ os.system("nvidia-smi -L")
10
 
11
  model_map = {'Van Gogh' : 'models/vangogh.pt',
12
  'Pablo Picasso': 'models/pablopicasso.pt',