Update app.py

app.py

@@ -11,5 +11,5 @@ text = st.text_input("Enter the starting text:", "Replace me by any text you'd l
 if st.button("Generate"):
     encoded_input = tokenizer(text, return_tensors='pt', padding=False, truncation=False)
     output = model(**encoded_input)
-    generated_text = output
+    generated_text = tokenizer.decode(torch.argmax(output[0], dim=1), skip_special_tokens=True)
     st.write(generated_text)