Update app.py

app.py

@@ -25,8 +25,8 @@ last_hidden_states = outputs.last_hidden_state
 
 output = last_hidden_states
 
-decoded = tokenizer.decode(output[0][0]) # [0][0]
-print(decoded)
+# decoded = tokenizer.decode(output[0][0]) # [0][0]
+# print(decoded)
 
 # output = Seq2SeqModelOutput(output)
 
@@ -35,7 +35,7 @@ print(decoded)
 with st.container():
     st.write('\n\n')
     st.write('LLM-LANAChat\n\n')
-    st.write(decoded)
+    st.write(output)
 
 print('\nsaida gerada.')
 print('\n\n')