Update app.py

app.py

@@ -14,8 +14,8 @@ def next_word(text, pipe):
   }
     res=pipe(text)
     for e in res:
-        res_dict['token_str'].extend(e['token_str'])
-        res_dict['score'].extend(e['score'])
+        res_dict['token_str'].append(e['token_str'])
+        res_dict['score'].append(e['score'])
     return res_dict
 
 st.title("Predict Next Word")