UPDATE

app.py

@@ -8,9 +8,12 @@ def generate_image(prompt, style_id):
     try:
         styles = Somnium.Styles()
         image = Somnium.Generate(prompt, styles[style_id])
-        return image
+        if image == None:
+            raise gr.Error("Process contains NSFW")
+        else:
+            return image
     except Exception as e:
-        raise gr.Error("Process failed or contains NSFW")
+        raise gr.Error("Process failed")
 
 # create interface
 iface = gr.Interface(