Update app.py

app.py

@@ -36,21 +36,21 @@ sp = gr.ScatterPlot(x="x", y="y", color="color")
 def inference(word1, word2, word3):
     output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
     word_list = [word1, word2, word3]
-    word_list.extend([x for x,y in output[:4]])
+    word_list.extend([x for x,y in [item for item in output[:4]]])
     words = {key: model[key] for key in word_list}
     data = np.concatenate([x[np.newaxis, :] for x in words.values()], axis=0)
     print(data.shape)
     labels = words.keys()
     reduced_data_pca = reduce_dimensions(data, method='PCA')
     print(reduced_data_pca.shape)
-    '''
+    #'''
     df = pd.DataFrame({
         "x": reduced_data_pca[:, 0], 
         "y": reduced_data_pca[:, 1],
         "color": ["b", "b", "b", "r", "g", "g", "g"]
     })
-    '''
-    return None #df
+    #'''
+    return df
 
 examples = [
     ["woman", "man", "aunt"],