Update summarizer.py

summarizer.py

@@ -1,9 +1,7 @@
-import streamlit as st
-from transformers import pipeline
+# use bart in pytorch
+summarizer = pipeline("summarization")
+summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)
 
-pipe = pipeline('summarization')
-text = st.text_area('enter some text!')
-
-if text:
-  out = pipe(text)
-  st.json(out)
+# use t5 in tf
+summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
+summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)