Spaces:
Runtime error
Runtime error
| # --- | |
| # jupyter: | |
| # jupytext: | |
| # formats: ipynb,py:percent | |
| # text_representation: | |
| # extension: .py | |
| # format_name: percent | |
| # format_version: '1.3' | |
| # jupytext_version: 1.13.8 | |
| # kernelspec: | |
| # display_name: Python 3 (ipykernel) | |
| # language: python | |
| # name: python3 | |
| # --- | |
| # %% | |
| import gradio as gr | |
| from fastai.text.all import * | |
| from nltk import tokenize | |
| import nltk | |
| nltk.download("punkt") | |
| # %% | |
| # model_pkl = "https://cloudstor.aarnet.edu.au/plus/s/rJzjZMk7ieZGsao/download" | |
| # p = Path("./formality.pkl") | |
| # %% | |
| learner_pkl_url = "https://cloudstor.aarnet.edu.au/plus/s/lRDFdt5KmkhxFr6/download" | |
| p = Path("./learner.pkl") | |
| # %% | |
| if not p.exists(): | |
| import urllib.request | |
| urllib.request.urlretrieve(learner_pkl_url, "./learner.pkl") | |
| # %% | |
| # m = load_pickle("./formality.pkl") | |
| # %% | |
| formal_sentences = "The Gram matrix (or Gramian matrix, Gramian) of a set of vectors in an inner product space is the Hermitian matrix of inner products. C'man, man, don't leave holding the bag!" | |
| # %% | |
| m = load_learner(p) | |
| # %% | |
| L(tokenize.sent_tokenize(formal_sentences)).map(lambda x: m.predict(x)[0]) | |
| # %% tags=[] | |
| def formal_informal(text): | |
| # a = m.predict(text) | |
| # assessment = "style: {0}, confidence {1:.2f}%".format( a[0], a[2][0] * 100) | |
| sentences = L(tokenize.sent_tokenize(text)) | |
| predictions = sentences.map(lambda x: m.predict(x)[0]) | |
| # assessment = "style: {0}".format(a[0]) | |
| assessment = zip(sentences.map(lambda x: x + ' '), predictions) | |
| return assessment | |
| demo = gr.Interface(formal_informal, | |
| gr.Textbox(label="Your text", lines=5, value=""), | |
| gr.HighlightedText(label="Stylistic analysis", color_map={'formal': 'blue', 'informal': 'yellow'}, | |
| combine_adjacent=True, show_legend=False), | |
| examples=["I returned and saw under the sun, that the race is not to the swift, nor the battle to the strong, neither yet bread to the wise, nor yet riches to men of understanding, nor yet favour to men of skill; but time and chance happeneth to them all.", | |
| "Objective consideration of contemporary phenomena compels the conclusion that success or failure in competitive activities exhibits no tendency to be commensurate with innate capacity, but that a considerable element of the unpredictable must invariably be taken into account.", | |
| "When the process of freeing a vehicle that has been stuck results in ruts or holes, the operator will fill the rut or hole created by such activity before removing the vehicle from the immediate area.", | |
| "If you make a hole while freeing a stuck vehicle, you must fill the hole before you drive away.", | |
| "Prior to surgery patients are required to cleanse with this fluid.", | |
| "The day of your surgery, wash your body with this liquid soap."]) | |
| # First two examples are from Orwell's "Politics and the English Language" | |
| # Next two examples are from https://www.plainlanguage.gov/examples/before-and-after/wordiness/ | |
| # Next two examples are from https://www.ncbi.nlm.nih.gov/pmc/articles/PMC6044302/ | |
| demo.launch() | |
| # %% | |