Upload folder using huggingface_hub

app.py

@@ -284,8 +284,13 @@ def generate_image(
             if status == "completed":
                 elapsed = time.time() - start_time
                 output_url = data["outputs"][0]
-                yield f"🎉 Generation successful! Time taken {elapsed:.1f}s", output_url, output_url, update_recent_gallery(
-                    prompt, input_image, output_url)
+                has_nsfw_content = data["has_nsfw_contents"][0]
+                if has_nsfw_content:
+                    error_message = "❌ NSFW content detected in the output"
+                    yield error_message, None, "", None
+                else:
+                    yield f"🎉 Generation successful! Time taken {elapsed:.1f}s", output_url, output_url, update_recent_gallery(
+                        prompt, input_image, output_url)
                 return
             elif status == "failed":
                 raise Exception(data.get("error", "Unknown error"))