Spaces:
Runtime error
Runtime error
import gradio as gr | |
from memsum import MemSum | |
sent_tokenize = lambda txt: txt.split('. ') | |
model_path = "model/MemSum_Final/model.pt" | |
summarizer = MemSum(model_path, "model/glove/vocabulary_200dim.pkl") | |
def preprocess(text): | |
text = text.replace("\n","") | |
text = text.replace("¶","") | |
text = " ".join(text.split()) | |
return text | |
def summarize(text): | |
text = sent_tokenize( preprocess(text) ) | |
summary = "\n".join( summarizer.summarize(text) ) | |
return summary | |
demo = gr.Interface(fn=summarize, inputs="text", outputs="text") | |
demo.launch() |