Upload app.py

app.py

@@ -12,24 +12,15 @@ from heapq import nlargest
 import warnings
 from sklearn.feature_extraction.text import TfidfVectorizer
 import numpy as np
-#from transformers import PegasusForConditionalGeneration, PegasusTokenizer
 
 warnings.filterwarnings("ignore")
 
+pegasus = gr.Interface.load("huggingface/google/pegasus-xsum") 
+
 def get_wiki_original_text(inp):
     text = wikipedia.summary(inp)
     return text
 
-"""
-def  get_wiki_summary_by_pegasus(inp):
-    text = wikipedia.summary(inp)
-    tokenizer = PegasusTokenizer.from_pretrained('google/pegasus-xsum')
-    tokens = tokenizer(text, truncation=True, padding="longest", return_tensors="pt")
-    model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
-    summary = model.generate(**tokens)
-    return tokenizer.decode(summary)
-    """
-
 
 def get_wiki_summary_by_lem(inp):
     text = wikipedia.summary(inp)
@@ -111,21 +102,16 @@ def get_wiki_summary_by_tfidf(inp):
 
 
 
-desc =  """This interface allows you to summarize Wikipedia explanations. Only requirement is to write the topic. For summarization this model uses extractive summarization method and the number of sentences in the output depends on the length of the original text."""
-
-  
-x = """ Europe """
-
-y = ''' Great Depression '''
+desc =  """This interface allows you to summarize Wikipedia contents. Only requirement is to write the topic and it collects content by fetching from Wikipedia. For summarization this model uses 2 different extractive summarization methods and the number of sentences in the output depends on the length of the original text."""
 
-z = ''' Crocodile Dundee '''
 
-sample = [[x],[y],[z]]
+sample = [['Europe'],['Great Depression'],['Crocodile Dundee']]
 
 
 iface = Parallel(gr.Interface(fn=get_wiki_original_text, inputs=gr.inputs.Textbox(label="Text"), outputs="text", description='Original Text'),
                  gr.Interface(fn=get_wiki_summary_by_lem, inputs=gr.inputs.Textbox(label="Text"), outputs="text", description='Summary 1'),
                  gr.Interface(fn=get_wiki_summary_by_tfidf, inputs=gr.inputs.Textbox(label="Text"), outputs="text", description='Summary 2'),
+                 pegasus,
                  title= 'Text Summarizer', 
                  description = desc,
                  examples=sample,