small fixes

pages/2_Concepts_comparison.py

@@ -1,6 +1,9 @@
 import streamlit as st
 import streamlit.components.v1 as components
 
+import dnnlib
+import legacy
+
 import pickle
 import pandas as pd
 import numpy as np
@@ -155,7 +158,7 @@ with output_col_2:
                 print(f'Similarity between {c1} and {c2}')
                 similarity = cosine_similarity(vectors_CLIP[c1].reshape(1, -1), vectors_CLIP[c2].reshape(1, -1))
                 print(np.round(similarity[0][0], 3))
-                edges_clip.append((c1, c2, float(np.round(similarity[0][0], 3))))
+                edges_clip.append((c1, c2, np.round(float(np.round(similarity[0][0], 3))), 3))
 
 
     net_clip = Network(height="750px", width="100%",)