tkasasagi commited on
Commit
e856705
1 Parent(s): 0780ea6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ MAX_SEED = np.iinfo(np.int32).max
29
  device = "cuda" if torch.cuda.is_available() else "cpu"
30
 
31
  NUM_IMAGES_PER_PROMPT = 1
32
- SAFETY_CHECKER = False
33
  if SAFETY_CHECKER:
34
  from safety_checker import StableDiffusionSafetyChecker
35
  from transformers import CLIPFeatureExtractor
 
29
  device = "cuda" if torch.cuda.is_available() else "cpu"
30
 
31
  NUM_IMAGES_PER_PROMPT = 1
32
+ SAFETY_CHECKER = True
33
  if SAFETY_CHECKER:
34
  from safety_checker import StableDiffusionSafetyChecker
35
  from transformers import CLIPFeatureExtractor