Update app.py

app.py

@@ -25,7 +25,7 @@ def code_similarity(code1, code2, model):
     else:
         v1 = uae.encode(code1)
         v2 = uae.encode(code2)
-    return cosine_similarity(v1, v2)
+    return cosine_similarity(v1[0], v2[0])
 
 
 demo = gr.Interface(