Prakash N commited on
Commit
088c6b5
1 Parent(s): 691e068

modified 1st summary button functionality

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,7 +35,7 @@ st.markdown(""" #### LorSor helps you through a simple 3 stage process.
35
  st.markdown('#')
36
  col1, col2 = st.columns(2)
37
 
38
- @st.cache
39
  def get_response(input_text):
40
  batch = tokenizer([input_text],truncation=True,padding='longest',max_length=1024, return_tensors="pt").to('cpu')
41
  gen_out = model.generate(**batch,max_length=128,num_beams=5, num_return_sequences=1, temperature=1.5)
@@ -58,4 +58,4 @@ with col2:
58
  summary = "<< Add some text in ( Step 1 ) for me to summarize >>"
59
  y = st.text_area("Here is the completed summary for you to edit", summary)
60
  st.button("Submit edits")
61
- st.balloons()
 
35
  st.markdown('#')
36
  col1, col2 = st.columns(2)
37
 
38
+ # @st.cache
39
  def get_response(input_text):
40
  batch = tokenizer([input_text],truncation=True,padding='longest',max_length=1024, return_tensors="pt").to('cpu')
41
  gen_out = model.generate(**batch,max_length=128,num_beams=5, num_return_sequences=1, temperature=1.5)
 
58
  summary = "<< Add some text in ( Step 1 ) for me to summarize >>"
59
  y = st.text_area("Here is the completed summary for you to edit", summary)
60
  st.button("Submit edits")
61
+ # st.balloons()