sairamn commited on
Commit
c18799d
·
1 Parent(s): 8734fdd

Add app.py and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import streamlit as st
2
  from transformers import BartTokenizer, TFBartForConditionalGeneration
3
- model_name = 'facebook-bart-large-cnn'
 
4
  tokenizer = BartTokenizer.from_pretrained(model_name)
5
  model = TFBartForConditionalGeneration.from_pretrained(model_name)
6
 
@@ -17,4 +18,4 @@ if st.button('Summarize'):
17
  summary = summarize(user_input)
18
  st.write(summary)
19
  else:
20
- st.write("Please enter some text to summarize.")
 
1
  import streamlit as st
2
  from transformers import BartTokenizer, TFBartForConditionalGeneration
3
+
4
+ model_name = 'facebook/bart-large-cnn'
5
  tokenizer = BartTokenizer.from_pretrained(model_name)
6
  model = TFBartForConditionalGeneration.from_pretrained(model_name)
7
 
 
18
  summary = summarize(user_input)
19
  st.write(summary)
20
  else:
21
+ st.write("Please enter some text to summarize.")
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
  tensorflow==2.17.0
2
  transformers==4.32.0
3
- keras==2.18.0
4
  streamlit==1.24.0
 
1
  tensorflow==2.17.0
2
  transformers==4.32.0
 
3
  streamlit==1.24.0