Update app.py

app.py

@@ -30,7 +30,7 @@ output = last_hidden_states[0][0]
 
 # output = Seq2SeqModelOutput(output)
 
-# output = tokenizer.batch_decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
+output = tokenizer.batch_decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
 
 with st.container():
     st.write('\n\n')