Duplicate from dominguesm/positive-reframing-en

Co-authored-by: Maicon Domingues <dominguesm@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Positive Reframing EN
+emoji: 💻
+colorFrom: purple
+colorTo: gray
+sdk: gradio
+sdk_version: 3.9.1
+app_file: app.py
+pinned: false
+duplicated_from: dominguesm/positive-reframing-en
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,72 @@
+from difflib import Differ
+
+import gradio as gr
+from transformers import pipeline
+
+pipe = pipeline("summarization", "dominguesm/positive-reframing-en")
+
+
+def predict(text, operation):
+
+    try:
+        res = pipe(f"[{operation}]: {text}", max_length=124)
+    except Exception as e:
+        return e
+
+    d = Differ()
+    return (
+        res[0]["summary_text"],
+        [
+            (token[2:], token[0] if token[0] != " " else None)
+            for token in d.compare(text, res[0]["summary_text"])
+        ],
+    )
+    # return res[0]["summary_text"]
+
+
+iface = gr.Interface(
+    title="Positive Reframing EN",
+    description="This model is a T5 adjusted to the sentiment transfer task, where the objective is to reverse the sentiment polarity of a text without contradicting the original meaning. Positive reframing induces a complementary positive viewpoint (e.g. glass-half-full) escaping negative patterns. More info [here](https://huggingface.co/dominguesm/positive-reframing-en).",
+    fn=predict,
+    inputs=[
+        gr.Textbox(
+            lines=1,
+            placeholder=(
+                f"Pensar no meu futuro me faz querer viver numa ilha sozinha para sempre"
+            ),
+        ),
+        gr.Radio(
+            [
+                "growth",
+                "impermanence",
+                "neutralizing",
+                "optimism",
+                "self_affirmation",
+                "thankfulness",
+            ]
+        ),
+    ],
+    outputs=[
+        gr.Textbox(label="Generated Text"),
+        gr.HighlightedText(
+            label="Diff",
+            combine_adjacent=True,
+        ).style(color_map={"+": "green", "-": "red"}),
+    ],
+    examples=[
+        [
+            "You know I really don't care about the power struggle between the papacy and secular authority in the medieval ages. stupid",
+            "growth",
+        ],
+        [
+            "thinking about my future makes me want to go live on a island alone forever. annoyed",
+            "optimism",
+        ],
+        [
+            "Who would have ever guessed that it would be so freaking hard to get three different grades from two different schools together.",
+            "thankfulness",
+        ],
+    ],
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1 @@
+transformers[torch]