First commit

Makefile

@@ -0,0 +1,15 @@
+install:
+	pip install --upgrade pip &&\
+		pip install -r requirements.txt
+
+debugthree:
+	# not working as expect
+	python -m pytest -vv --pdb --maxfail=4 # drop to pdb first three failures
+
+format:
+	black *.py
+
+lint:
+	pylint --disable=R,C *.py
+
+all: install lint test format

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+import torch
+from transformers import BertTokenizerFast, EncoderDecoderModel
+
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+ckpt = 'mrm8488/bert2bert_shared-spanish-finetuned-summarization'
+tokenizer = BertTokenizerFast.from_pretrained(ckpt)
+model = EncoderDecoderModel.from_pretrained(ckpt).to(device)
+
+def generate_summary(text):
+    inputs = tokenizer([text], padding="max_length", truncation=True, max_length=512, return_tensors="pt")
+    input_ids = inputs.input_ids.to(device)
+    attention_mask = inputs.attention_mask.to(device)
+    output = model.generate(input_ids, attention_mask=attention_mask)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
+
+
+textbox = gr.Textbox(placeholder="Introduzca texto a resumir ...", lines=4)
+demo = gr.Interface(fn=generate_summary, inputs=textbox,outputs="text")
+
+demo.launch()

flagged/log.csv

@@ -0,0 +1,13 @@
+text,output,flag,username,timestamp
+"Manuel Pellegrini aclaró lesión de Claudio Bravo
+El entrenador de Real Betis se refirió a la ausencia del arquero en el partido contra Villarreal.
+
+Manuel Pellegrini aclaró lesión de Claudio Bravo
+AlAireLibre.clLLÉVATELO:
+El entrenador chileno Manuel Pellegrini contó detalles sobre la lesión que aqueja al fubolista Claudio Bravo, quien quedó al margen del partido en que Real Betis derrotó 2-1 a Villarreal.
+
+El DT nacional comentó además cuánto tiempo se estima que el golero formado en Colo Colo se mantenga fuera de las canchas.
+
+""Claudio está con un problema en el sóleo, no es un problema muy grande, espero que en una o dos semanas esté recuperado"", declaró.
+
+Real Betis derrotó a Villarreal, equipo en que milita Ben Brereton (que se quedó en el banco de suplentes), por la primera fecha de la liga española.",El técnico de Real Betis se refiere a la ausencia del arquero en el partido contra Villarreal,,,2023-08-14 00:03:01.072542

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers
+tensorflow
+torch