Update app.py

app.py

@@ -7,7 +7,7 @@ from transformers import pipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
-st.title('Marcel Proust')
+st.title('Booba')
 st.subheader("Commencez la phrase, l'algorithme la termine.")
 st.write("Note : la génération du texte prend ~ 5 minutes")
 
@@ -16,7 +16,7 @@ st.write("Note : la génération du texte prend ~ 5 minutes")
 
 with st.form("my_form"):
    
-   text = st.text_input("Début de la phrase :", 'Je me souvenais de ce jour où Jean-Michel')
+   text = st.text_input("Début de la phrase :", "C'était Noël dans la famille")
    
    # Every form must have a submit button.
    submitted = st.form_submit_button("Générer la suite")
@@ -24,7 +24,7 @@ with st.form("my_form"):
    # Load the model ---
    model_checkpoint = "bigscience/bloom-560m"
    tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
-   model = AutoModelForCausalLM.from_pretrained("dan-vdb/ProustAI")
+   model = AutoModelForCausalLM.from_pretrained("dan-vdb/BoobaAI")
    device = torch.device("cpu")
 
    # device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")