Upload app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from gradio.mix import Parallel, Series
 import wikipedia
 import spacy
 from spacy.lang.en.stop_words import STOP_WORDS
@@ -9,13 +10,29 @@ nltk.download('punkt', quiet=True)
 from nltk.stem import WordNetLemmatizer
 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")
 
-def get_wiki_summary(inp):
+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)
-    print('\033[1m' + "Original Text Fetched from Wikipedia" + '\033[0m')
 
     print(text)
 
@@ -61,5 +78,26 @@ def get_wiki_summary(inp):
 
     return summary
 
-if __name__ == '__main__':
-    gr.Interface(fn=get_wiki_summary, inputs=gr.inputs.Textbox(label="Requested Topic from Wikipedia    :   "), outputs="text").launch(inline=False, share=True)
+
+
+
+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."""
+
+
+sample = [['Europe'], ['Great Depression'], ['Crocodile Dundee']]
+
+
+iface = Parallel(gr.Interface(fn=get_wiki_original_text, inputs=gr.inputs.Textbox(label="Requested Topic from Wikipedia    :   "), outputs="text"),
+                 gr.Interface(fn=get_wiki_summary_by_lem, inputs=gr.inputs.Textbox(label="Requested Topic from Wikipedia    :   "), outputs="text"),
+                 gr.Interface(fn=get_wiki_summary_by_pegasus, inputs=gr.inputs.Textbox(label="Requested Topic from Wikipedia    :   "), outputs="text"),
+    
+    
+    
+    
+  #  get_wiki_original_text,get_wiki_summary_by_lem, get_wiki_summary_by_pegasus,
+                 title= 'Text Summarizer', 
+                 description = desc,
+                 examples=sample, 
+                 inputs = gr.inputs.Textbox(label="Text"))
+
+iface.launch(inline = False)