fffiloni commited on
Commit
020d346
1 Parent(s): dd965ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,8 +40,8 @@ def infer(prompt):
40
  words = is_safe.split()
41
  # Take the first word and convert it to a string variable
42
  status = str(words[0])
43
- if status == "True" :
44
- raise gr.Error("Don't.")
45
  else:
46
  results = pipe(prompt, guidance_scale=7.5)
47
  #for i in range(len(results.images)):
 
40
  words = is_safe.split()
41
  # Take the first word and convert it to a string variable
42
  status = str(words[0])
43
+ if status == "Yes" :
44
+ raise gr.Error("Don't ask for such things.")
45
  else:
46
  results = pipe(prompt, guidance_scale=7.5)
47
  #for i in range(len(results.images)):