Spaces:
Runtime error
Runtime error
Commit
·
7bf73ed
1
Parent(s):
079d713
Update app.py
Browse files
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 topo -m")
|
| 10 |
|
| 11 |
model_map = {'Van Gogh' : 'models/vangogh.pt',
|
| 12 |
'Pablo Picasso': 'models/pablopicasso.pt',
|