Update app.py

app.py

@@ -104,7 +104,7 @@ class Visualizer:
  if d <= thresh and (neighbor == entity2 or neighbor == entity1) and len(neighbors) < num_neighbors + 1:
  neighbors.add(entity1)
  neighbors.add(entity2)
- spring_tension = (thresh - d) * 100 # for vis purposes
+ spring_tension = ((thresh - d) * 20000) ** 2 # for vis purposes
  G.add_edge(entity1, entity2, weight=spring_tension)
  neighbors.remove(neighbor)
  for entity1, entity2, d in tqdm(normalized_distances):
@@ -121,7 +121,7 @@ class Visualizer:
  # nx.draw_networkx_edges(G, pos, edgelist=edges_not_connected_to_specific_node, edge_color='gray', alpha=0.1, width=1)
  for u, v, d in edges:
  if u == neighbor or v == neighbor:
- nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 100)) * 10, 2)}, font_color="red", alpha=0.4) # reverse modifications
+ nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 20000) ** (1 / 2)), 2) * 10}, font_color="red", alpha=0.4) # reverse modifications
  nx.draw_networkx_labels(G, pos, font_size=14, font_family='sans-serif', font_color='green')
  nx.draw_networkx_labels(G, pos, labels={neighbor: neighbor}, font_size=14, font_family='sans-serif', font_color='red')
  plt.title(f'Graph of {distance_type}')
@@ -131,7 +131,7 @@ class Visualizer:
 
 
 if __name__ == '__main__':
- vis = Visualizer(cache_root="Preprocessing/multilinguality")
+ vis = Visualizer(cache_root=".")
  text_selection = [f"{vis.iso_codes_to_names[iso_code]}" for iso_code in vis.iso_codes_to_names]
  iface = gr.Interface(fn=vis.visualize,
  inputs=[gr.Dropdown(["Physical Distance between Language Centroids on the Globe",