com3dian commited on
Commit
d459820
·
verified ·
1 Parent(s): 23cb72b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -42,11 +42,11 @@ with col2:
42
  max_value=100,
43
  value=(25, 75)
44
  )
 
45
 
46
- summ_text = None
47
 
48
- if uploaded_file is not None:
49
-
50
  st.write(uploaded_file.name)
51
  bytes_data = uploaded_file.getvalue()
52
  st.write(len(bytes_data), "bytes")
@@ -76,10 +76,8 @@ if uploaded_file is not None:
76
  summarizor = pipeline("summarization", model=summ_model_path, device = device)
77
  summ_text = summarizor(document.segmentation['text'], max_length=100, min_length=10, do_sample=False)
78
  summ_text = [text['summary_text'] for text in summ_text]
79
-
80
- for summ in summ_text:
81
- st.write(summ)
82
-
83
 
84
  if summ_text is not None:
85
 
 
42
  max_value=100,
43
  value=(25, 75)
44
  )
45
+ st.session_state.generation_done = False
46
 
 
47
 
48
+ if uploaded_file is not None and not st.session_state.generation_done:
49
+ summ_text = None
50
  st.write(uploaded_file.name)
51
  bytes_data = uploaded_file.getvalue()
52
  st.write(len(bytes_data), "bytes")
 
76
  summarizor = pipeline("summarization", model=summ_model_path, device = device)
77
  summ_text = summarizor(document.segmentation['text'], max_length=100, min_length=10, do_sample=False)
78
  summ_text = [text['summary_text'] for text in summ_text]
79
+
80
+ st.session_state.generation_done = True
 
 
81
 
82
  if summ_text is not None:
83