Update app.py

app.py

@@ -21,21 +21,22 @@ model = XGLMModel.from_pretrained("facebook/xglm-564M")
 inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model(**inputs) #, labels=inputs["input_ids"])
 
+
 last_hidden_states = outputs.last_hidden_state
+logits = last_hidden_states.logits
 
-output = last_hidden_states[0][0]
+# output = last_hidden_states[0][0]
 
 # decoded = tokenizer.decode(output[0][0]) # [0][0]
 # print(decoded)
 
 # output = Seq2SeqModelOutput(output)
 
-output = tokenizer.batch_decode(output, 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')
     st.write('LLM-LANAChat\n\n')
-    st.write(output)
+    st.write(logits)
 
 print('\nsaida gerada.')
 print('\n\n')