Jinl commited on
Commit
2b37c45
·
1 Parent(s): eac4c16

add NSFW checker and GPU mode

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  import torch
3
  import random
4
  import numpy as np
@@ -17,7 +18,7 @@ from torchvision.utils import save_image
17
  import utils.ptp_utils as ptp_utils
18
 
19
  import torchvision.transforms as transforms
20
- import spaces
21
 
22
  try:
23
  import xformers
 
1
  import os
2
+ import spaces
3
  import torch
4
  import random
5
  import numpy as np
 
18
  import utils.ptp_utils as ptp_utils
19
 
20
  import torchvision.transforms as transforms
21
+
22
 
23
  try:
24
  import xformers