Update app.py

app.py

@@ -93,19 +93,31 @@ class Measurer:
 
  def get_dists(self, l1, l2):
  if l1 in self.tree_dist_func:
- tree_dist = self.tree_dist_func[l1][l2]
+ if l2 in self.tree_dist_func[l1]:
+ tree_dist = self.tree_dist_func[l1][l2]
+ else:
+ tree_dist = self.tree_dist_func[l2][l1]
  else:
  tree_dist = self.tree_dist_func[l2][l1]
  if l1 in self.map_dist_func:
- map_dist = self.map_dist_func[l1][l2]
+ if l2 in self.map_dist_func[l1]:
+ map_dist = self.map_dist_func[l1][l2]
+ else:
+ map_dist = self.map_dist_func[l2][l1]
  else:
  map_dist = self.map_dist_func[l2][l1]
  if l1 in self.asp_dist_func:
- asp_dist = self.asp_dist_func[l1][l2]
+ if l2 in self.asp_dist_func[l1]:
+ asp_dist = self.asp_dist_func[l1][l2]
+ else:
+ asp_dist = self.asp_dist_func[l2][l1]
  else:
  asp_dist = self.asp_dist_func[l2][l1]
  if l1 in self.learned_dist_func:
- learned_dist = self.learned_dist_func[l1][l2]
+ if l2 in self.learned_dist_func[l1]:
+ learned_dist = self.learned_dist_func[l1][l2]
+ else:
+ learned_dist = self.learned_dist_func[l2][l1]
  else:
  learned_dist = self.learned_dist_func[l2][l1]
  return tree_dist, map_dist, asp_dist, learned_dist
@@ -142,5 +154,6 @@ iface = gr.Interface(fn=m.measure,
  outputs=[gr.Plot(),
  gr.Plot(),
  gr.Plot(),
- gr.Plot()])
+ gr.Plot()],
+ article="This demo allows you to view the distance between two languages from the ISO 639-3 list according to several distance measurement functions. For more information, check out our paper: https://arxiv.org/abs/2406.06403 and our text-to-speech tool, in which we make use of this technique: https://github.com/DigitalPhonetics/IMS-Toucan")
 iface.launch()