E-Adam commited on
Commit
3702fa3
·
verified ·
1 Parent(s): 9b3a5df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -74,9 +74,9 @@ def apprendre(liste_actuelle):
74
 
75
  # fonction pour prédire le mot suivant
76
  def predict_next_word(start_text):
77
- next_words = 5 if start_text.count(" ")<5 else 3
78
- print(f"prédictions de {nb} mots.")
79
- for _ in range(next_words):
80
  token_list = tokenizer.texts_to_sequences([start_text])[0]
81
  token_list = pad_sequences([token_list], maxlen=max_sequence_len-1, padding='pre')
82
  predicted = np.argmax(model.predict(token_list), axis=-1)
 
74
 
75
  # fonction pour prédire le mot suivant
76
  def predict_next_word(start_text):
77
+ nb_words = 5 if start_text.count(" ")<5 else 3
78
+ print(f"prédictions de {nb_words} mots.")
79
+ for _ in range(nb_words):
80
  token_list = tokenizer.texts_to_sequences([start_text])[0]
81
  token_list = pad_sequences([token_list], maxlen=max_sequence_len-1, padding='pre')
82
  predicted = np.argmax(model.predict(token_list), axis=-1)