Update app.py

app.py

@@ -30,9 +30,11 @@ def find_similar_documents(query, documents, model_name):
     sorted_indices = torch.argsort(similarities, descending=True)
     results = []
     for i, idx in enumerate(sorted_indices):
-        score = similarities[idx].item()
-        results.append(f"**{i+1}.** ({score:.4f}) — {doc_list[idx]}")
-    return "\n".join(results)
+        score = similarities[0][idx].item()
+        doc = doc_list[idx]
+        results.append(f"{i+1}. Score: {score:.4f}\n   Document: {doc}")
+    
+    return "\n\n".join(results)
 
 def compare_models(query, documents, tarka_model, open_model):
     tarka_result = find_similar_documents(query, documents, tarka_model)