Update app.py

app.py

@@ -47,7 +47,7 @@ def train(data: str, message: str):
     
     model.fit(X, y, epochs=10, batch_size=8, workers=4, use_multiprocessing=True)
     tokens = tokenizer.texts_to_sequences([message,])[0]
-    prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))
+    prediction = model.predict(np.array([(list(tokens)+[0,]*inp_len)[:inp_len],]))[0]
     max_o = 0
     max_v = 0
     for ind, i in enumerate(prediction):