handle oov error

app.py

@@ -24,7 +24,11 @@ def find_most_similar(word):
   return format_to_html(sim_output)
 
 def find_similarity(wordA, wordB):
-  return max([float(wv.similarity(wordA, wordB)), 0])
+  try:
+    sim = wv.similarity(wordA, wordB)
+  except Exception:
+    return "[Error] OOV"
+  return max([float(sim), 0])
 
 title = "Check Word Similarity With Literary Word2Vec"
 examples = ["སེམས་", "བླ་མ་", "སྟོབས་"]
@@ -34,7 +38,7 @@ print(find_similarity(examples[0], examples[1]))
 demo = gr.Interface(
   fn=find_similarity,
   inputs=[gr.inputs.Textbox("Word A"), gr.inputs.Textbox("Word B")],
-  outputs="number",
+  outputs="text",
   title=title,
   # examples=examples
 ).launch()