Update app.py

app.py

@@ -184,6 +184,7 @@ def image_click(images, evt: gr.SelectData,
     #### corr_array
     corr_array = corr_array_df.loc[cols].iloc[:, :top_k].values
     l_list = pd.Series(corr_array.reshape([-1])).values.tolist()
+    l_list = pd.Series(l_list).map(Image.open).map(lambda x: expand2square(x, (0, 0, 0))).values.tolist()
     l_dist_list = []
     for ele in l_list:
         if ele not in l_dist_list:
@@ -223,4 +224,5 @@ with gr.Blocks() as demo:
         outputs = [sdg_outputs, sg_outputs],)
 
 
+
 demo.launch("0.0.0.0")