chore: lable interface

app.py

@@ -23,20 +23,19 @@ examples=[
 def get_titles(query):
     query_embed = model.encode(query)
     hits = semantic_search(query_embed, title_embeddings, top_k=5)[0]
-    
-    titles = list()
-
+    titles = dict()
     for hit in hits:
         index = hit["corpus_id"]
         selected_title = title_dataset["train"]["title"][index]
-        # score = hit["score"]
-        titles.append(selected_title)
-    return "\n".join(titles)
+        score = hit["score"]
+        titles[selected_title] = score
+    return titles
 
 space = gr.Interface(
     fn=get_titles,
     inputs=gr.Textbox(label="Input Title"),
-    outputs=gr.Textbox(label="Similar Titles"),
+    # outputs=gr.Textbox(label="Similar Titles"),
+    outputs=gr.Label(num_top_classes=5)
     title=title,
     description=description,
     examples=examples,