Update app.py

app.py

@@ -3,20 +3,24 @@ import spacy
 from spacy import displacy
 from spacy.tokens import Span
 from random import randint
-from fastcoref import FCoref
-
-model = FCoref("pythainlp/han-coref-v1.0")
+import spacy
+from fastcoref import spacy_component
 nlp = spacy.blank("th")
 
+nlp.add_pipe(
+   "fastcoref", 
+   config={'model_architecture': 'FCoref','model_path': 'pythainlp/han-coref-v1.0'}
+)
+
 default = "แกว่าเราเข้มแข็งมากพอที่จะเห็นแกไปน่ารักกับใครหรอ"
 
 def corefer(text):
-    preds = model.predict(texts=[text])
-    clusters = preds[0].get_clusters(as_strings=False)
+    doc = nlp(text)
+    tags = doc._.coref_clusters
     title=None
     dic_ents = {"text":text,"ents":[],"title":title}
-    _tag=[str(i) for i in list(range(len(clusters)))]
-    for i,tag in enumerate(clusters):
+    _tag=[str(i) for i in list(range(len(tags)))]
+    for i,tag in enumerate(tags):
         for s,e in tag:
             dic_ents["ents"].append({"start": s, "end": e, "label": _tag[i]})
     colors={i:"#"+''.join([random.choice('0123456789ABCDEF') for j in range(6)]) for i in _tag} # thank https://stackoverflow.com/a/50218895