Moran commited on
Commit
de7955a
·
1 Parent(s): e3ddc6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,9 +28,9 @@ def start_summarize(long_text, model):
28
  else:
29
  if model=="Bart":
30
  repo_id = "Aviv/Moran_Aviv_Bart"
31
- summary = "It works!"
32
- # inf_learn = from_pretrained_fastai(repo_id)
33
- # summary = inf_learn.blurr_generate([long_text])#[0]['generated_texts']
34
 
35
  else:
36
  summary = long_text
 
28
  else:
29
  if model=="Bart":
30
  repo_id = "Aviv/Moran_Aviv_Bart"
31
+ # summary = "It works!"
32
+ inf_learn = from_pretrained_fastai(repo_id)
33
+ summary = inf_learn.blurr_generate([long_text][0]['generated_texts']
34
 
35
  else:
36
  summary = long_text