Soyoung97 commited on
Commit
77149e4
1 Parent(s): 0882f0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -41,5 +41,8 @@ if text:
41
  num_beams=4,
42
  early_stopping=True,
43
  repetition_penalty=2.0)
44
- summ = tokenizer.decode(corrected_ids.squeeze().tolist(), skip_special_tokens=True)
 
 
 
45
  st.write(summ)
 
41
  num_beams=4,
42
  early_stopping=True,
43
  repetition_penalty=2.0)
44
+ output = tokenizer.decode(corrected_ids.squeeze().tolist(), skip_special_tokens=True)
45
+ if output == '':
46
+ output = 'Nothing generated...TT Please try again with different text!'
47
+
48
  st.write(summ)