Update app.py
Browse files
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 == "
|
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)):
|