Update app.py

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 reranker = CrossEncoder("sentence-transformers/all-MiniLM-L12-v2")
 
 
-def rerank_documents(query: str, documents: pd.DataFrame) -> pd.DataFrame:
+def rerank(query: str, documents: pd.DataFrame) -> pd.DataFrame:
     documents = documents.copy()
     documents = documents.drop_duplicates("text")
     documents["rank"] = reranker.predict([[query, hit] for hit in documents["text"]])
@@ -33,7 +33,7 @@ with gr.Blocks() as demo:
     )
 
     submit_btn.click(
-        fn=rerank_documents,
+        fn=rerank,
         inputs=[query_input, documents_input],
         outputs=[documents_output],
     )