Update app.py

app.py

@@ -13,13 +13,14 @@ default = "แกว่าเราเข้มแข็งมากพอที
 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 )
+    title=None
+    dic_ents = {"text":text,"ents":[],"title":title}
+    _tag=[str(i) for i in list(range(len(clusters)))]
+    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
+    return displacy.render(dic_ents, manual=True, style="ent",options=colors)
 
 
 iface = gr.Interface(fn=corefer,