Update app.py

app.py

@@ -3,9 +3,9 @@ import streamlit as st
 
 from transformers import AutoTokenizer, AutoModelWithLMHead
   
-tokenizer = AutoTokenizer.from_pretrained("gpt2-large")
+tokenizer = AutoTokenizer.from_pretrained("mrm8488/spanish-gpt2")
 
-model = AutoModelWithLMHead.from_pretrained("gpt2-large")
+model = AutoModelWithLMHead.from_pretrained("mrm8488/spanish-gpt2")
 
 
 
@@ -26,7 +26,7 @@ def infer(input_ids, max_length, temperature, top_k, top_p, num_return_sequences
 default_value = "See how a modern neural network auto-completes your text 🤗 This site, built by the Hugging Face team, lets you write a whole document directly from your browser, and you can trigger the Transformer anywhere using the Tab key. Its like having a smart machine that completes your thoughts 😀 Get started by typing a custom snippet, check out the repository, or try one of the examples. Have fun!"
 
 #prompts
-st.title("Write with Transformers 🦄")
+st.title("Write with Spanish GPT-2 🦄")
 st.write("The almighty king of text generation, GPT-2 comes in four available sizes, only three of which have been publicly made available. Feared for its fake news generation capabilities, it currently stands as the most syntactically coherent model. A direct successor to the original GPT, it reinforces the already established pre-training/fine-tuning killer duo. From the paper: Language Models are Unsupervised Multitask Learners by Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei and Ilya Sutskever.")
 
 sent = st.text_area("Text", default_value, height = 275)