from typing import Any, Dict
import gradio as gr
from pipeline import LongCorefPipeline
from huggingface_hub import snapshot_download

long_coref = LongCorefPipeline(snapshot_download("kwang2049/long-coref"))


def fn(doc: str) -> Dict[str, Any]:
    return long_coref(doc)


demo = gr.Interface(fn=fn, inputs="text", outputs="text")
demo.launch()