Spaces:

Soooma
/

titles_gen

Runtime error

Soooma commited on Aug 1, 2023

Commit

4583b4b

1 Parent(s): dd52464

minor change

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,6 +21,6 @@ if text:
     decoded_output = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
     predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
-    html_string = f"<h3>The predicted title is:</h3> \'{predicted_title}\'"
     st.markdown(html_string, unsafe_allow_html=True)

     decoded_output = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
     predicted_title = nltk.sent_tokenize(decoded_output.strip())[0]
+    html_string = f"<h4>The predicted title is:</h4> \'{predicted_title}\'"
     st.markdown(html_string, unsafe_allow_html=True)