Duplicate from pythiccoder/FastCoref

Co-authored-by: Ari Bornstein <[email protected]>

.gitattributes

@@ -0,0 +1,31 @@
+*.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
+*.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
+*.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
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb 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
+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,16 @@
+---
+title: FastCoref
+models:
+- biu-nlp/lingmess-coref
+emoji: 📚
+colorFrom: pink
+colorTo: blue
+sdk: gradio
+sdk_version: 3.2
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: pythiccoder/FastCoref
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import spacy
+from spacy import displacy
+from spacy.tokens import Span
+from random import randint
+from fastcoref import LingMessCoref
+
+model = LingMessCoref()
+nlp = spacy.blank("en")
+
+default = "Lionel Messi has won a record seven Ballon d'Or awards. He signed for Paris Saint-Germain in August 2021. “I would like to thank my family” said the Argentinian footballer. Messi holds the records for most goals in La Liga. Paris Saint-Germain hopes he will do the same in Ligue 1."
+
+def corefer(text):
+    preds = model.predict(texts=[text])
+    clusters = preds[0].get_clusters(as_strings=False)
+    doc = nlp(text)
+    doc.spans["sc"] = []
+    colors = {"Cluster {}".format(i):'#%06X' % randint(0, 0xFFFFFF) for i in range(len(clusters))}
+    for i, cluster in enumerate(clusters):
+      for sp in cluster:
+        doc.spans["sc"] += [doc.char_span(sp[0], sp[1], "Cluster {}".format(i))]
+    return displacy.render(doc, style="span", options= {"colors":colors }, page=True )
+
+
+iface = gr.Interface(fn=corefer, 
+                     inputs=gr.Textbox(label="Enter Text To Corefer with FastCoref", lines=2, value=default), 
+                     outputs="html")
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+spacy==3.4.1
+fastcoref==1.7.1