Update app.py

app.py

@@ -1,13 +1,14 @@
 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"]
+  #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"]
 
 with gr.Blocks() as demo:
   gr.Markdown("""# Summarization Engine!""")
@@ -23,7 +24,7 @@ with gr.Blocks() as demo:
         gr.Textbox(label="Attractiveness Score", default="hot"),
     ]
       b1 = gr.Button("Summarize!")
-      b1.click(main, inputs=[text1], outputs=["Holy fuck","Holy fucking fuck","That body of yours is insane","hot"], api_name="s")
+      b1.click(main, inputs=[text1], outputs=output, api_name="s")