Spaces:
Runtime error
Runtime error
Commit
·
e0bcdbd
1
Parent(s):
bc70d86
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,8 @@ from StableDiffuser import StableDiffuser
|
|
| 5 |
from tqdm import tqdm
|
| 6 |
from train import train
|
| 7 |
|
| 8 |
-
|
|
|
|
| 9 |
|
| 10 |
model_map = {'Van Gogh' : 'models/vangogh.pt',
|
| 11 |
'Pablo Picasso': 'models/pablopicasso.pt',
|
|
|
|
| 5 |
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',
|