Initial commit

README.md

@@ -1,37 +1,11 @@
 ---
-title: Cyberbullying Detector
-emoji: 🏃
+title: CyberBullying Detector
+emoji: 😭
 colorFrom: purple
 colorTo: indigo
 sdk: gradio
 app_file: app.py
-pinned: false
+pinned: true
 ---
 
-# Configuration
 
-`title`: _string_  
-Display title for the Space
-
-`emoji`: _string_  
-Space emoji (emoji-only character allowed)
-
-`colorFrom`: _string_  
-Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
-
-`colorTo`: _string_  
-Color for Thumbnail gradient (red, yellow, green, blue, indigo, purple, pink, gray)
-
-`sdk`: _string_  
-Can be either `gradio` or `streamlit`
-
-`sdk_version` : _string_  
-Only applicable for `streamlit` SDK.  
-See [doc](https://hf.co/docs/hub/spaces) for more info on supported versions.
-
-`app_file`: _string_  
-Path to your main application file (which contains either `gradio` or `streamlit` Python code).  
-Path is relative to the root of the repository.
-
-`pinned`: _boolean_  
-Whether the Space stays on top of your list.

app.py

@@ -0,0 +1,39 @@
+import gradio as gr
+import re
+
+from transformers import pipeline
+
+sp_model = "JonatanGk/roberta-base-bne-finetuned-cyberbullying-spanish"
+ca_model = "JonatanGk/roberta-base-ca-finetuned-cyberbullying-catalan"
+sp_analysis = pipeline("text-classification", model=sp_model, tokenizer=sp_model)
+ca_analysis = pipeline("text-classification", model=ca_model, tokenizer=ca_model)
+
+def bullying_analysis(language, text):
+    if language == 'Spanish':
+        results = sp_analysis(text)
+    elif language == 'Catalan':
+        results = ca_analysis(text)
+    return results[0]["label"], round(results[0]["score"], 5)
+
+
+gradio_ui = gr.Interface(
+    fn=bullying_analysis,
+    title="CyberBullying Detector (Spanish/Catalan)",
+    description="Enter some text and check if model detects bullying.",
+    inputs=[
+        gr.inputs.Radio(['Spanish','Catalan'],label='Language',),
+        gr.inputs.Textbox(lines=5, label="Paste some text here"),
+    ],
+    outputs=[
+        gr.outputs.Textbox(label="Label"),
+        gr.outputs.Textbox(label="Score"),
+    ],
+    examples=[
+        ['Spanish', "Eres mas alto que un pino y mas tonto que un pepino!"],
+        ['Catalan', "Ets un barrufet!"],
+        ['Spanish', "Estas mas gordo que una foca!"],
+        ['Catalan', "Ets mes lleig que un pecat!"],
+    ],
+)
+
+gradio_ui.launch()

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch