Update app.py

app.py

@@ -6,7 +6,7 @@ model = CrossEncoder("jinaai/jina-reranker-v2-base-multilingual", trust_remote_c
 
 # Function to rerank documents
 def rerank(query, documents):
-    documents = documents.split("\n")  # Split input into a list
+    documents = documents.split("|||")  # Use custom delimiter to split documents correctly
     scores = model.predict([[query, doc] for doc in documents if doc.strip()])
     ranked_docs = sorted(zip(documents, scores), key=lambda x: x[1], reverse=True)
     return [{"document": doc, "score": round(score, 4)} for doc, score in ranked_docs]
@@ -14,10 +14,10 @@ def rerank(query, documents):
 # Gradio Interface
 iface = gr.Interface(
     fn=rerank,
-    inputs=["text", gr.Textbox(label="Documents (One per line)", lines=5, placeholder="Enter one document per line")], 
+    inputs=["text", gr.Textbox(label="Documents (Separate with |||)", placeholder="Doc1 ||| Doc2 ||| Doc3")], 
     outputs="json",
     title="JinaAI v2 Reranker API",
-    description="Enter a query and a list of documents. The model will rank them based on relevance.",
+    description="Enter a query and documents (separated by '|||'). The model will rank them based on relevance.",
 )
 
 iface.launch()