Changing to paste in text for input since the wikipedia api doesn't work.

app.py

@@ -3,15 +3,16 @@ import wikipedia
 from transformers import pipeline
 import os
 
-## Setting to use the 0th GPU
+# Setting to use the 0th GPU
 os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
+
 def summarize(text):
-    ## Setting to use the bart-large-cnn model for summarization
+    # Setting to use the bart-large-cnn model for summarization
     summarizer = pipeline("summarization")
 
-    ## To use the t5-base model for summarization:
-    ## summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
+    # To use the t5-base model for summarization:
+    # summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
 
     summary_text = summarizer(text, max_length=100, min_length=5, do_sample=False)[0]['summary_text']
     print(f'Length of initial text: {len(text)}')
@@ -19,6 +20,7 @@ def summarize(text):
     print(summary_text)
     return summary_text
 
+
 def greet(name):
     return "Hello " + name.orig_name + "!!"
 
@@ -32,9 +34,9 @@ def search_wiki(text):
 
 
 def get_wiki(search_term):
-    text = wikipedia.summary(search_term)
-    orig_text_len = len(text)
-    text = summarize(text)
+    # text = wikipedia.summary(search_term)
+    orig_text_len = len(search_term)
+    text = summarize(search_term)
     sum_length = len(text)
     return [text, orig_text_len, sum_length]
 
@@ -49,9 +51,9 @@ out_sum_text_len = gr.Number(label='Summarized Text Length')
 
 iface = gr.Interface(fn=get_wiki,
                      inputs=gr.Textbox(lines=50, placeholder="Wikipedia search term here...", label='Search Term'),
-                     outputs=[out_sum_text,out_orig_test_len,out_sum_text_len],
+                     outputs=[out_sum_text, out_orig_test_len, out_sum_text_len],
                      title='Article Summary',
                      description='Paste in an article and it will be summarized',
                      sample_inputs='guardians of the galaxy'
-)
-iface.launch()  # To create a public link, set `share=True` in `launch()`.
+                     )
+iface.launch()  # To create a public link, set `share=True` in `launch()`.