ki33elev commited on
Commit
5492f24
1 Parent(s): ce8e0e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -52,6 +52,7 @@ tokenizer, model = load_model()
52
 
53
  title = st.text_area(label='Title', height=100)
54
  summary = st.text_area(label='Summary (optional)', height=250)
 
55
 
56
  prediction, prediction_probs = predict(title, summary, tokenizer, model)
57
  ans = get_results(prediction, prediction_probs)
 
52
 
53
  title = st.text_area(label='Title', height=100)
54
  summary = st.text_area(label='Summary (optional)', height=250)
55
+ button = st.button('Run')
56
 
57
  prediction, prediction_probs = predict(title, summary, tokenizer, model)
58
  ans = get_results(prediction, prediction_probs)