Return first result

app.py

@@ -23,7 +23,7 @@ def search(query):
     cross_inp = [[query, hit] for hit in hits]
     cross_scores = cross_encoder.predict(cross_inp)
     hits = [hit for _, hit in sorted(zip(cross_scores, hits), reverse=True)]
-    return hits
+    return hits[0]
 
 iface = gr.Interface(fn=search, inputs="text", outputs="text")
 iface.launch()