Spaces:

amielle
/

patent-summarizer

Runtime error

amielle commited on Mar 22, 2022

Commit

926f2cd

•

1 Parent(s): 8ecadd9

chore: Add model initialization

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model_names = ["google/bigbird-pegasus-large-bigpatent",
                "huggingface/facebook/bart-large-cnn",
                "huggingface/google/pegasus-xsum"]
 patent_summarizer = summarizer.PatentSummarizer(model_collection)
 iface = gr.Interface(
     patent_summarizer.pipeline,

                "huggingface/facebook/bart-large-cnn",
                "huggingface/google/pegasus-xsum"]
+model_collection = summarizer.init_models()
 patent_summarizer = summarizer.PatentSummarizer(model_collection)
 iface = gr.Interface(
     patent_summarizer.pipeline,