Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import nltk
 import math
 import torch
 
@@ -82,7 +81,7 @@ def generate_title():
     # compute predictions
     outputs = model.generate(**inputs, do_sample=True, temperature=temperature, max_new_tokens=64)
     decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
-    predicted_titles = [nltk.sent_tokenize(decoded_output.strip())[0] for decoded_output in decoded_outputs]
+    predicted_titles = decoded_output
 
     st.session_state.titles = predicted_titles