Update app.py

app.py

@@ -33,7 +33,7 @@ def plot_reduced_data(reduced_data, labels, title):
                  arrowprops=dict(arrowstyle="->", color='green', lw=3))
     
     # Data for the arrow 2
-    end_point = (reduced_data[2, 0] - (reduced_data[0, 0] - reduced_data[1, 0]), reduced_data[2, 1] - (reduced_data[0, 1] - reduced_data[1, 1]))  # Starting point of the arrow
+    end_point = ((reduced_data[-1, 0] , (reduced_data[-1, 1]))  # Starting point of the arrow
     start_point = (reduced_data[2, 0], reduced_data[2, 1])  # Ending point of the arrow
     
     # Adding an arrow 2
@@ -64,11 +64,13 @@ sp = gr.Image()
 
 
 def inference(word1, word2, word3):
-    output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
+    transform = model[word3] + model[word2] - model[word1]
+    output = model.similar_by_vector(transform)
     print(output)
     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
     data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
     print(data.shape)
     labels = words.keys()