update

app.py

@@ -83,8 +83,8 @@ def generate_title():
     # compute predictions
     outputs = model.generate(**inputs, do_sample=True, temperature=temperature)
     decoded_outputs = tokenizer.batch_decode(outputs, skip_special_tokens=True)
-    # predicted_titles = [decoded_output[0] for decoded_output in decoded_outputs]
-    predicted_titles = [nltk.sent_tokenize(decoded_output.strip())[0] for decoded_output in decoded_outputs]
+    predicted_titles = [decoded_output for decoded_output in decoded_outputs]
+    # predicted_titles = [nltk.sent_tokenize(decoded_output.strip())[0] for decoded_output in decoded_outputs]
     
     st.session_state.titles = predicted_titles