Update app.py

app.py

@@ -83,12 +83,12 @@ 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 = [nltk.sent_tokenize(decoded_output.strip())[0] for decoded_output in decoded_outputs]
-
+    predicted_titles = [decoded_output[0] for decoded_output in decoded_outputs]
+    
     st.session_state.titles = predicted_titles
 
 # generate title button
-st_generate_button = st.button('Generate query', on_click=generate_title)
+st_generate_button = st.button('生成query', on_click=generate_title)
 
 # title generation labels
 if 'titles' not in st.session_state: