Update app.py

app.py

@@ -26,13 +26,15 @@ last_hidden_states = outputs.last_hidden_state
 
 output = last_hidden_states #['last_hidden_states']
 
+# decoded = tokenizer.decode(output)
+
 # output = BaseModelOutput(last_hidden_states['last_hidden_states'])
 
 # logits = last_hidden_states.logits
 
 # output = last_hidden_states[0][0]
 
-decoded = tokenizer.decode(output) # [0][0]
+# decoded = tokenizer.decode(output) # [0][0]
 # print(decoded)
 
 # output = Seq2SeqModelOutput(output)
@@ -42,7 +44,7 @@ with st.container():
     st.write('\n\n')
     st.write('LLM-LANAChat\n\n')
     st.write(output)
-    st.write(decoded)
+    st.write(type(output))
 
 print('\nsaida gerada.')
 print('\n\n')