Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
-#from transformers import pipeline
+from transformers import pipeline
 
-#pipe = pipeline("summarization", model="google/bigbird-pegasus-large-arxiv")
+pipe = pipeline("summarization", model="google/bigbird-pegasus-large-arxiv")
 
 def main(in_text):
   print(in_text)
-  #answer = pipe(in_text, min_length=5, max_length=round(len(in_text)/4))
-  #print(answer)
-  #return answer[0]["summary_text"]
-  return ["Holy fuck","Holy fucking fuck","That body of yours is insane","hot"]
+  answer = pipe(in_text, min_length=5, max_length=round(len(in_text)/4))
+  print(answer)
+  return answer[0]["summary_text"]
 
 with gr.Blocks() as demo:
   gr.Markdown("""# Summarization Engine!""")
@@ -18,15 +17,9 @@ with gr.Blocks() as demo:
             label="Input Text",
             lines=1,
         )
-      output = [gr.Textbox(label="Total Hot or Not™ Score", default="Holy fuck"),
-        gr.Textbox(label="Hotness Score", default="Holy fucking fuck"),
-        gr.Textbox(label="Beauty Score", default="That body of yours is insane"),
-        gr.Textbox(label="Attractiveness Score", default="hot"),
-    ]
+      output = gr.Textbox(label="Output Text")
       b1 = gr.Button("Summarize!")
       b1.click(main, inputs=[text1], outputs=output, api_name="s")
-
-