Update app.py

app.py

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