Update app.py

app.py

@@ -22,7 +22,7 @@ def plot_reduced_data(reduced_data, labels, title):
     plt.figure(figsize=(10, 8))
     plt.scatter(reduced_data[:, 0], reduced_data[:, 1], alpha=0.6)
     for i, label in enumerate(labels):
-        plt.annotate(" " + label, (reduced_data[i, 0], reduced_data[i, 1]), fontsize=14)
+        plt.annotate("  " + label, (reduced_data[i, 0], reduced_data[i, 1]), fontsize=18)
     plt.title(title)
     # Data for the arrow 1
     start_point = (reduced_data[0, 0], reduced_data[0, 1])  # Starting point of the arrow
@@ -70,7 +70,7 @@ def inference(word1, word2, word3):
     word_list = [word1, word2, word3]
     word_list.extend([x for x,y in [item for item in output[:4]]])
     words = {key: model[key] for key in word_list}
-    words["Equivalent"] = transform
+    words[word3 + " + (" + word2 + " - " + word1 + ")"] = transform
     data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
     print(data.shape)
     labels = words.keys()