Update app.py

app.py

@@ -24,7 +24,6 @@ def topk_candidates(query, candidates, k):
     topk = wl.topk(query, candidates, k)
     return topk
 
-# Define the Gradio interface
 def create_gradio_interface():
     with gr.Blocks() as demo:
         gr.Markdown("## WordLlama Gradio Demo")
@@ -43,7 +42,7 @@ def create_gradio_interface():
         with gr.Tab("Rank Documents"):
             query = gr.Textbox(label="Query", placeholder="Enter the query here...")
             candidates = gr.Textbox(label="Candidates (comma separated)", placeholder="Enter candidate sentences here...")
-            ranked_docs_output = gr.Dataframe(headers=["Document", "Score"], datatype=["str", "float"])
+            ranked_docs_output = gr.Dataframe(headers=["Document", "Score"])
             gr.Button("Rank Documents").click(
                 fn=lambda q, c: rank_documents(q, c.split(',')),
                 inputs=[query, candidates],
@@ -86,4 +85,4 @@ def create_gradio_interface():
 
 # Create and launch the Gradio interface
 demo = create_gradio_interface()
-demo.launch(debug = True)
+demo.launch()