Spaces:
Running
on
Zero
Running
on
Zero
| import html | |
| import gradio as gr | |
| import spaces | |
| from transformers import pipeline | |
| # Load the pipeline (token classification) | |
| #token_classifier = pipeline("token-classification", model="WesScivetti/SNACS_English", aggregation_strategy="simple") | |
| # <-- required for ZeroGPU | |
| def classify_tokens(text): | |
| color_dict = {'None': '#6adf97', | |
| 'O': '#f18621', | |
| 'p.Purpose-p.Purpose': '#554065', | |
| 'p.SocialRel-p.Gestalt': '#8ea0d7', | |
| 'B-p.Cost-p.Cost': '#f4b518', | |
| 'p.Topic-p.Topic': '#976cae', | |
| 'p.Originator-p.Gestalt': '#f25ca8', | |
| 'p.Originator-p.Source': '#a08323', | |
| 'p.Recipient-p.Goal': '#725be0', | |
| 'p.Possessor-p.Possessor': '#b5ce9e', | |
| 'p.Gestalt-p.Gestalt': '#34a8a9', | |
| 'p.Ancillary-p.Ancillary': '#73f29f', | |
| 'p.ComparisonRef-p.Goal': '#6a26db', | |
| 'p.Source-p.Source': '#5cc334', | |
| 'p.Theme-p.Theme': '#5b88c8', | |
| 'p.Locus-p.Locus': '#4c39c8', | |
| 'p.Characteristic-p.Characteristic': '#661943', | |
| 'p.Explanation-p.Explanation': '#852e58', | |
| 'p.OrgMember-p.Possessor': '#e3bd42', | |
| 'p.Goal-p.Goal': '#6bfc3c', | |
| 'p.Manner-p.Manner': '#436097', | |
| 'p.ComparisonRef-p.ComparisonRef': '#4df5a9', | |
| 'p.Cost-p.Locus': '#fe5990', | |
| 'p.Duration-p.Duration': '#5e454e', | |
| 'p.Identity-p.Identity': '#cb49ed', | |
| 'p.OrgMember-p.Gestalt': '#18fdd1', | |
| 'p.Experiencer-p.Goal': '#400043', | |
| 'p.QuantityItem-p.Whole': '#5f3ba4', | |
| 'p.Whole-p.Gestalt': '#497114', | |
| 'p.PartPortion-p.PartPortion': '#edfc14', | |
| 'p.Time-p.Time': '#4605b0', | |
| 'p.Approximator-p.Approximator': '#553ee1', | |
| 'p.Direction-p.Direction': '#687447', | |
| 'p.Locus-p.Direction': '#12b336', | |
| 'p.Instrument-p.Path': '#0ccdda', | |
| 'p.QuantityItem-p.Gestalt': '#d88be2', | |
| 'p.Species-p.Species': '#4dfc63', | |
| 'p.Org-p.Ancillary': '#6a5b9c', | |
| 'p.Agent-p.Gestalt': '#f373bf', | |
| 'p.SocialRel-p.Ancillary': '#4ee1dc', | |
| 'p.Circumstance-p.Locus': '#38abe5', | |
| 'p.Circumstance-p.Circumstance': '#69caeb', | |
| 'p.Whole-p.Whole': '#00d816', | |
| 'p.QuantityItem-p.QuantityItem': '#dbbc2d', | |
| 'p.Theme-p.Purpose': '#cb56ba', | |
| 'p.Goal-p.Locus': '#b3597f', | |
| 'p.Extent-p.Extent': '#5cadfa', | |
| 'p.Experiencer-p.Gestalt': '#8275f4', | |
| 'p.Means-p.Means': '#b1bfb7', | |
| 'p.Beneficiary-p.Beneficiary': '#0e9582', | |
| 'p.Org-p.Beneficiary': '#c48ea7', | |
| 'p.Stimulus-p.Topic': '#a6af3a', | |
| 'p.Recipient-p.Ancillary': '#a5ff4b', | |
| 'p.Beneficiary-p.Possessor': '#c941dc', | |
| 'p.Agent-p.Ancillary': '#d18ce9', | |
| 'p.Theme-p.Gestalt': '#b71c4f', | |
| 'p.StartTime-p.StartTime': '#9b3cf9', | |
| 'p.Cost-p.Extent': '#117f70', | |
| 'p.Manner-p.Source': '#460233', | |
| 'p.Characteristic-p.Source': '#41c518', | |
| 'p.Locus-p.Path': '#d3c136', | |
| 'p.Manner-p.ComparisonRef': '#32cbcb', | |
| 'p.Extent-p.Whole': '#94454f', | |
| 'p.Experiencer-p.Beneficiary': '#1f2d98', | |
| 'p.Theme-p.ComparisonRef': '#ef3f97', | |
| 'p.Stuff-p.Stuff': '#9919e8', | |
| 'p.Theme-p.Goal': '#d7c6d1', | |
| 'p.Interval-p.Interval': '#042206', | |
| 'p.Time-p.Whole': '#ecf0a1', | |
| 'p.Stimulus-p.Beneficiary': '#af168a', | |
| 'p.Characteristic-p.Locus': '#ac54e6', | |
| 'p.Characteristic-p.Extent': '#0ec04c', | |
| 'p.EndTime-p.EndTime': '#29e89e', | |
| 'p.Experiencer-p.Ancillary': '#bce155', | |
| 'p.Agent-p.Agent': '#aac43b', | |
| 'p.PartPortion-p.Source': '#9eb3c3', | |
| 'p.Locus-p.Source': '#7121d7', | |
| 'p.Duration-p.Extent': '#ca1096', | |
| 'p.Characteristic-p.Identity': '#345c8d', | |
| 'p.Possession-p.PartPortion': '#e592aa', | |
| 'p.Possession-p.Theme': '#a59bec', | |
| 'p.Whole-p.Locus': '#0bc209', | |
| 'p.Direction-p.Goal': '#9d90cd', | |
| 'p.Gestalt-p.Locus': '#97f830', | |
| 'p.Org-p.Gestalt': '#2f2c3c', | |
| 'p.Stimulus-p.Goal': '#c40f02', | |
| 'p.Theme-p.Instrument': '#a312ed', | |
| 'p.Stimulus-p.Force': '#d98ddb', | |
| 'p.Beneficiary-p.Theme': '#68fdb4', | |
| 'p.Characteristic-p.Goal': '#a60b97', | |
| 'p.Time-p.Goal': '#97567c', | |
| 'p.Explanation-p.Time': '#90f72f', | |
| 'p.Instrument-p.Manner': '#2b1869', | |
| 'p.Possession-p.Ancillary': '#a9672c', | |
| 'p.Instrument-p.Instrument': '#6eb1ef', | |
| 'p.Ensemble-p.Ancillary': '#93fb41', | |
| 'p.Recipient-p.Gestalt': '#0674a2', | |
| 'p.Agent-p.Source': '#bf427f', | |
| 'p.Whole-p.Source': '#dae5cb', | |
| 'p.Stimulus-p.Explanation': '#108bd6', | |
| 'p.Stimulus-p.Direction': '#aa0f64', | |
| 'p.ComparisonRef-p.Purpose': '#65fb63', | |
| 'p.ComparisonRef-p.Locus': '#e48da2', | |
| 'p.Theme-p.Ancillary': '#685b19', | |
| 'p.Identity-p.ComparisonRef': '#caac20', | |
| 'p.QuantityItem-p.Stuff': '#a1f649', | |
| 'p.Recipient-p.Direction': '#a8ba9d', | |
| 'p.Path-p.Locus': '#03c408', | |
| 'p.Originator-p.Agent': '#b46878', | |
| 'p.Beneficiary-p.Gestalt': '#26eaf0', | |
| 'p.Possessor-p.Ancillary': '#dd8d5e', | |
| 'p.Beneficiary-p.Goal': '#212bd7', | |
| 'p.OrgMember-p.PartPortion': '#bd7620', | |
| 'p.PartPortion-p.ComparisonRef': '#6fd197', | |
| 'p.Frequency-p.Extent': '#8a9e22', | |
| 'p.Beneficiary-p.Direction': '#094599', | |
| 'p.Characteristic-p.Stuff': '#02889c', | |
| 'p.Manner-p.Extent': '#686d06', | |
| 'p.Cost-p.Cost': '#f4b518', | |
| 'p.Theme-p.Whole': '#5a51fb', | |
| 'p.Frequency-p.Frequency': '#d26bc7', | |
| 'p.Purpose-p.Locus': '#80e1ac', | |
| 'p.Force-p.Gestalt': '#1063d3', | |
| 'p.Characteristic-p.Ancillary': '#947622', | |
| 'p.ComparisonRef-p.Source': '#b0954c', | |
| 'p.Org-p.Instrument': '#e2bfce', | |
| 'p.Theme-p.Characteristic': '#44b67f', | |
| 'p.Characteristic-p.Topic': '#b90264', | |
| 'p.Locus-p.Goal': '#5d62c0', | |
| 'p.Locus-p.Whole': '#e4222b', | |
| 'p.Theme-p.Locus': '#60211c', | |
| 'p.Frequency-p.Manner': '#6b5831', | |
| 'p.Locus-p.Ancillary': '#8de37d', | |
| 'p.Topic-p.Identity': '#10a385', | |
| 'p.Org-p.Goal': '#b42090', | |
| 'p.SetIteration-p.SetIteration': '#11e7a6', | |
| 'p.PartPortion-p.Goal': '#ee8159', | |
| 'p.ComparisonRef-p.Ancillary': '#3270a9', | |
| 'p.Force-p.Force': '#dc6a3a', | |
| 'p.Approximator-p.Extent': '#005d48', | |
| 'p.Manner-p.Stuff': '#920903', | |
| 'p.Path-p.Goal': '#543e80', | |
| 'p.Explanation-p.Source': '#e65656', | |
| 'p.Topic-p.Goal': '#31bcfc', | |
| 'p.Possession-p.Locus': '#1312e3', | |
| 'p.Circumstance-p.Path': '#8b9109', | |
| 'p.Gestalt-p.Source': '#7050ae', | |
| 'p.Agent-p.Locus': '#c9846e', | |
| 'p.Stimulus-p.Source': '#180a5f', | |
| 'p.Org-p.Whole': '#2a3053', | |
| 'p.Org-p.Source': '#ad1e85', | |
| 'p.Time-p.Extent': '#b1d4fa', | |
| 'p.Possessor-p.Locus': '#ae306d', | |
| 'p.Force-p.Source': '#727a29', | |
| 'p.Gestalt-p.Topic': '#f47f98', | |
| 'p.Cost-p.Manner': '#a61141', | |
| 'p.Means-p.Path': '#54d11a', | |
| 'p.Originator-p.Instrument': '#44fe8a', | |
| 'p.PartPortion-p.Instrument': '#4f7170', | |
| 'p.Possession-p.Possession': '#d3abe4', | |
| 'p.Agent-p.Beneficiary': '#1c515e', | |
| 'p.Instrument-p.Locus': '#4460b0', | |
| 'p.Instrument-p.Theme': '#1bed0b', | |
| 'p.Duration-p.Gestalt': '#2f787f', | |
| 'p.Path-p.Path': '#3637c0', | |
| 'p.Theme-p.Source': '#54a6f9', | |
| 'p.Time-p.Gestalt': '#24ff12', | |
| 'p.Time-p.Direction': '#9e135c', | |
| 'p.Goal-p.Whole': '#5fad91', | |
| 'p.Explanation-p.Manner': '#983754', | |
| 'p.Time-p.Interval': '#5cc4a8', | |
| 'p.Org-p.Locus': '#434851', | |
| 'p.Gestalt-p.Purpose': '#9ff474', | |
| 'p.Stimulus-p.Theme': '#12dfa1', | |
| 'p.Locus-p.Gestalt': '#636042', | |
| 'p.Extent-p.Identity': '#1414fd', | |
| 'p.ComparisonRef-p.Beneficiary': '#f47ef3', | |
| 'p.Experiencer-p.Agent': '#21883e', | |
| 'p.Time-p.Duration': '#98b42b', | |
| 'p.SocialRel-p.Source': '#4f3f8f', | |
| 'p.Whole-p.Circumstance': '#c70411', | |
| 'p.Purpose-p.Goal': '#f2f199'} | |
| token_classifier = pipeline("token-classification", model="WesScivetti/SNACS_English", | |
| aggregation_strategy="simple") | |
| results = token_classifier(text) | |
| sorted_results = sorted(results, key=lambda x: x["start"]) | |
| output = "" | |
| last_idx = 0 | |
| for entity in sorted_results: | |
| start = entity["start"] | |
| end = entity["end"] | |
| label = entity["entity_group"] | |
| score = entity["score"] | |
| word = html.escape(text[start:end]) | |
| output += html.escape(text[last_idx:start]) | |
| color = color_dict.get(label, "#D3D3D3") | |
| tooltip = f"{label} ({score:.2f})" | |
| output += ( | |
| f"<span style='background-color: {color}; padding: 2px; border-radius: 4px;' " | |
| f"title='{tooltip}'>{word}</span>" | |
| ) | |
| last_idx = end | |
| output += html.escape(text[last_idx:]) | |
| table = [ | |
| [entity["word"], entity["entity_group"], f"{entity['score']:.2f}"] | |
| for entity in sorted_results | |
| ] | |
| # Return both: HTML and table | |
| styled_html = f"<div style='font-family: sans-serif; line-height: 1.6;'>{output}</div>" | |
| return styled_html, table | |
| iface = gr.Interface( | |
| fn=classify_tokens, | |
| inputs=gr.Textbox(lines=4, placeholder="Enter a sentence...", label="Input Text"), | |
| outputs=[ | |
| gr.HTML(label="SNACS Tagged Sentence"), | |
| gr.Dataframe(headers=["Token", "SNACS Label", "Confidence"], label="SNACS Table") | |
| ], | |
| title="SNACS English Classification", | |
| description="SNACS English Classification. See the <a href='https://arxiv.org/abs/1704.02134'>SNACS guidelines</a> for details.", | |
| theme="default" | |
| ) | |
| iface.launch() |