Update app.py

app.py

@@ -48,7 +48,7 @@ def inference(word1, word2, word3):
     df = pd.DataFrame({
         "x": reduced_data_pca[:, 0], 
         "y": reduced_data_pca[:, 1],
-        "color": labels[:len(data)]
+        "color": [x for x in labels][:len(data)]
         #"label": ["W1", "W2", "W3", "W4", "W5", "W6", "W7"][:len(data)]
     })
     #'''