Co

app.py

@@ -140,11 +140,11 @@ def generate_hamiltonians(num_hamiltonians, selected_qubits, selected_order, wri
         store_in_duckdb(results_table)
 
 # Function to load results from either DuckDB or Hugging Face dataset
-def load_results(load_from_hf_checkbox, load_from_duckdb_checkbox):
-    if load_from_hf_checkbox:
-        return load_from_hf_dataset()  # Load from HF dataset
-    if load_from_duckdb_checkbox:
-        return load_from_duckdb()  # Load from DuckDB
+def load_results(load_from_hf, load_from_duckdb):
+    if load_from_hf:
+        return load_from_hf_dataset()
+    if load_from_duckdb:
+        return load_from_duckdb()
 
 # Gradio app
 with gr.Blocks() as app:
@@ -171,12 +171,12 @@ with gr.Blocks() as app:
         generate_button.click(update_status, inputs=[num_hamiltonians, selected_qubits, selected_order, write_to_hf, write_to_duckdb], outputs=status)
 
     with gr.Tab("View Results"):
-        load_from_hf_checkbox = gr.Checkbox(label="Load from Hugging Face dataset", value=False)
-        load_from_duckdb_checkbox = gr.Checkbox(label="Load from DuckDB", value=True)
+        load_from_hf = gr.Checkbox(label="Load from Hugging Face dataset", value=False)
+        load_from_duckdb = gr.Checkbox(label="Load from DuckDB", value=True)
 
         load_button = gr.Button("Load Results")
         output_display = gr.HTML()
 
-        load_button.click(load_results, inputs=[load_from_hf_checkbox, load_from_duckdb_checkbox], outputs=output_display)
+        load_button.click(load_results, inputs=[load_from_hf, load_from_duckdb], outputs=output_display)
 
-app.launch(share=True)
+app.launch()