Change layout to 2 columns

app.py

@@ -154,6 +154,7 @@ def predict_genus(method: str, dna_sequence: str, latitude: str, longitude: str)
     ax.set_title("Genus Prediction")
     ax.set_xlabel("Genus")
     ax.set_ylabel("Probability")
+    ax.set_xticks(range(len(top_k)))
     ax.set_xticklabels(top_k.index.astype(str), rotation=90)
     fig.subplots_adjust(bottom=0.3)
     fig.canvas.draw()
@@ -231,27 +232,21 @@ with gr.Blocks() as demo:
           that we precomputed and stored in a Pinecone index. Thie method
           DOES NOT examine ecological layer data.
         """)
-        # gr.Interface(
-        #     fn=predict_genus,
-        #     inputs=[
-        #         gr.Dropdown(choices=["cosine", "fine_tuned_model"], value="fine_tuned_model"),
-        #         inp_dna,
-        #         inp_lat,
-        #         inp_lng,
-        #     ],
-        #     outputs=["image"],
-        #     allow_flagging="never",
-        # )
-        
-        method_dropdown = gr.Dropdown(choices=["cosine", "fine_tuned_model"], value="fine_tuned_model")
-        predict_button = gr.Button("Predict Genus")
-        genus_output = gr.Image()
-        
-        predict_button.click(
-            fn=predict_genus,
-            inputs=[method_dropdown, inp_dna, inp_lat, inp_lng],
-            outputs=genus_output
-        )
+
+        with gr.Row():
+            with gr.Column():
+                method_dropdown = gr.Dropdown(
+                    choices=["cosine", "fine_tuned_model"], value="fine_tuned_model",
+                )
+                predict_button = gr.Button("Predict Genus")
+            with gr.Column():
+                genus_output = gr.Image()
+
+            predict_button.click(
+                fn=predict_genus,
+                inputs=[method_dropdown, inp_dna, inp_lat, inp_lng],
+                outputs=genus_output
+            )
 
     with gr.Tab("DNA Embedding Space Visualizer"):
         gr.Markdown("""
@@ -262,24 +257,20 @@ with gr.Blocks() as demo:
         learning to cluster similar DNA sequences together.
         """)
 
-        # gr.Interface(
-        #     fn=cluster_dna,
-        #     inputs=[
-        #         gr.Slider(minimum=1, maximum=10, step=1, value=5,
-        #                   label="Number of top genera to visualize")
-        #     ],
-        #     outputs=["image"],
-        #     allow_flagging="never",
-        # )
-
-        top_k_slider = gr.Slider(minimum=1, maximum=10, step=1, value=5, label="Number of top genera to visualize")
-        visualize_button = gr.Button("Visualize Embedding Space")
-        visualize_output = gr.Image()
-        
-        visualize_button.click(
-            fn=cluster_dna,
-            inputs=top_k_slider,
-            outputs=visualize_output
-        )
+        with gr.Row():
+            with gr.Column():
+                top_k_slider = gr.Slider(
+                    minimum=1, maximum=10, step=1, value=5,
+                    label="Number of top genera to visualize",
+                )
+                visualize_button = gr.Button("Visualize Embedding Space")
+            with gr.Column():
+                visualize_output = gr.Image()
+
+            visualize_button.click(
+                fn=cluster_dna,
+                inputs=top_k_slider,
+                outputs=visualize_output
+            )
 
 demo.launch()