Moran commited on
Commit
79a52b6
·
1 Parent(s): ad9aaf4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ def start_summarize(long_text, model):
51
  else:
52
  if model=="Bart":
53
  repo_id = "Aviv/Moran_Aviv_Bart"
54
- inf_learn = from_pretrained_fastai(repo_id)
55
- summary = textrank(long_text)[0]
56
  st.success(summary)
57
  time.sleep(10)
58
  grade = st.select_slider(
 
51
  else:
52
  if model=="Bart":
53
  repo_id = "Aviv/Moran_Aviv_Bart"
54
+ inf_learn = from_pretrained_fastai(repo_id)
55
+ summary = inf_learn.blurr_generate([text])
56
  st.success(summary)
57
  time.sleep(10)
58
  grade = st.select_slider(