Add application file

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 [7th]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

app.py

@@ -0,0 +1,167 @@
+import gradio as gr
+import timm
+import torch
+from transformers import RobertaForSequenceClassification, RobertaTokenizer
+
+"Vision"
+vit_model = timm.create_model("hf_hub:Marqo/nsfw-image-detection-384", pretrained=True)
+vit_model = vit_model.eval()
+
+data_config = timm.data.resolve_model_data_config(vit_model)
+transforms = timm.data.create_transform(**data_config, is_training=False)
+
+"NLP"
+tokenizer = RobertaTokenizer.from_pretrained("s-nlp/roberta_toxicity_classifier")
+model = RobertaForSequenceClassification.from_pretrained(
+    "s-nlp/roberta_toxicity_classifier"
+)
+
+
+def moderate_image(img):
+    # Load your model
+    with torch.no_grad():
+        output = vit_model(transforms(img).unsqueeze(0)).softmax(dim=-1).cpu()
+
+    class_names = vit_model.pretrained_cfg["label_names"]
+    probabilities = output[0].tolist()
+    if probabilities[0] >= 0.3:
+        return class_names[0]
+    else:
+        return class_names[1]
+
+
+def classify_toxic(text):
+    batch = tokenizer.encode(text, return_tensors="pt")
+    output = model(batch).cpu()
+    output = output.tolist()
+    max_idx = max(output)
+    return "Toxic" if output.index(max_idx) == 1 else "Neutral"
+
+
+# -----------------------
+# Apple-Minimal Styling
+# -----------------------
+custom_css = """
+/* Center container and control width */
+.gradio-container {
+    max-width: 900px !important;
+    margin: 0 auto !important;
+    padding: 20px 10px !important;
+}
+
+/* Header styling */
+.clean-title {
+    font-size: 1.9rem;
+    font-weight: 600;
+    text-align: center;
+    margin-bottom: 1.2rem;
+    letter-spacing: -0.4px;
+}
+
+/* Apple-like card sections */
+.apple-card {
+    padding: 18px;
+    border-radius: 12px;
+    border: 1px solid rgba(var(--block-border-color-rgb), 0.14);
+    background: var(--block-background-fill);
+    box-shadow: 0 1px 3px rgba(0,0,0,0.04);
+    margin-bottom: 18px;
+}
+
+/* Button styling: clean, flat, subtle */
+.gr-button {
+    border-radius: 8px !important;
+    background: var(--button-secondary-background-fill) !important;
+    border: 1px solid rgba(var(--block-border-color-rgb), 0.22) !important;
+    transition: 0.2s ease !important;
+}
+
+.gr-button:hover {
+    background: var(--button-secondary-background-fill-hover) !important;
+    border-color: rgba(var(--block-border-color-rgb), 0.34) !important;
+}
+
+.gr-button:active {
+    background: var(--button-secondary-background-fill-pressed) !important;
+}
+
+/* Reduce blank space between elements */
+.gr-block {
+    margin: 6px 0 !important;
+}
+
+/* Label style */
+label {
+    font-weight: 500 !important;
+}
+
+/* Make body fill full height so footer can stick */
+body, .gradio-container {
+    min-height: 100vh !important;
+    display: flex;
+    flex-direction: column;
+}
+
+/* Main content should expand, footer sits at bottom */
+.main-content {
+    flex: 1 0 auto;
+}
+
+.footer {
+    flex-shrink: 0;
+    text-align: center;
+    font-size: 0.85rem;
+    opacity: 0.6;
+    padding: 14px 0;
+    border-top: 1px solid rgba(var(--block-border-color-rgb), 0.12);
+    margin-top: 25px;
+}
+"""
+
+
+# -----------------------
+# UI Layout
+# -----------------------
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="violet", secondary_hue="slate"), css=custom_css
+) as demo:
+    with gr.Column(elem_classes="main-content"):
+        gr.Markdown("<div class='clean-title'>Content Safety Demo</div>")
+
+        with gr.Tabs():
+            # ---- NSFW Image Classification ---- #
+            with gr.Tab("NSFW Image Detection"):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        with gr.Group(elem_classes="apple-card"):
+                            img_in = gr.Image(type="pil", label="Upload Image")
+                            classify_img_btn = gr.Button("Classify")
+                            img_clear_btn = gr.ClearButton(components=img_in)
+
+                    with gr.Column(scale=2):
+                        with gr.Group(elem_classes="apple-card"):
+                            img_out = gr.Label(label="Prediction")
+
+                classify_img_btn.click(
+                    fn=moderate_image, inputs=img_in, outputs=img_out
+                )
+
+            # ---- Toxic Text Classification ---- #
+            with gr.Tab("Toxic Text Detection"):
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        with gr.Group(elem_classes="apple-card"):
+                            txt_in = gr.Textbox(lines=6, label="Enter Text")
+                            classify_txt_btn = gr.Button("Analyze")
+                            text_clear_btn = gr.ClearButton(components=txt_in)
+
+                    with gr.Column(scale=2):
+                        with gr.Group(elem_classes="apple-card"):
+                            txt_out = gr.Label(label="Prediction")
+
+                classify_txt_btn.click(classify_toxic, inputs=txt_in, outputs=txt_out)
+
+    gr.Markdown("<div class='footer'>Demo by 7th • Powered by Transformers</div>")
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+gradio
+transformers
+torch
+timm
+pillow