Initial commit

README.md

@@ -1,37 +1,11 @@
 ---
 title: Catalonia Independence Detector
-emoji: 👀
-colorFrom: indigo
+emoji: 👎👍
+colorFrom: yellow
 colorTo: pink
 sdk: gradio
 app_file: app.py
 pinned: false
 ---
 
-# 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-catalonia-independence-detector"
+ca_model = "JonatanGk/roberta-base-ca-finetuned-catalonia-independence-detector"
+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="Catalonia independence detector (Spanish/Catalan)",
+    description="Enter some text and check if model detects is favor/neutral/against Catalonia independence.",
+    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', "Junqueras, sobre la decisión judicial sobre Puigdemont: La justicia que falta en el Estado llega y llegará de Europa"],
+        ['Spanish', "Desconvocada la manifestación del domingo en Barcelona en apoyo a Puigdemont"],
+        ['Catalan', "Puigdemont, a l'estat espanyol: Quatre anys després, ens hem guanyat el dret a dir prou"],
+        ['Catalan', "Llarena demana la detenció de Comín i Ponsatí aprofitant que són a Itàlia amb Puigdemont"],
+    ],
+)
+
+gradio_ui.launch()

requirements.txt

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