Update app.py

app.py

@@ -68,7 +68,7 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
 
     # Set dark mode styling
     plt.style.use('dark_background')
-    plt.figure(figsize=(7, 7))
+    plt.figure(figsize=(10, 10))
     
     # Plot the confusion matrix with a dark-mode compatible colormap
     sns.heatmap(cm, cmap="magma", cbar=True, linecolor='white')
@@ -77,11 +77,11 @@ def plot_confusion_matrix_beamPred(cm, classes, title, save_path):
     plt.title(f"{title}\n(F1 Score: {avg_f1:.3f})", color="white", fontsize=14)
     
     tick_marks = np.arange(len(classes))
-    plt.xticks(tick_marks, classes, color="white", fontsize=12)  # White text for dark mode
-    plt.yticks(tick_marks, classes, color="white", fontsize=12)  # White text for dark mode
+    plt.xticks(tick_marks, classes, color="white", fontsize=14)  # White text for dark mode
+    plt.yticks(tick_marks, classes, color="white", fontsize=14)  # White text for dark mode
 
-    plt.ylabel('True label', color="white", fontsize=12)
-    plt.xlabel('Predicted label', color="white", fontsize=12)
+    plt.ylabel('True label', color="white", fontsize=14)
+    plt.xlabel('Predicted label', color="white", fontsize=14)
     plt.tight_layout()
     
     # Save the plot as an image
@@ -476,8 +476,8 @@ with gr.Blocks(css="""
                 task_complexity_dropdown = gr.Dropdown(label="Task Complexity (Number of Beams)", choices=[16, 32, 64, 128, 256], value=16)
 
         with gr.Row():
-            raw_img_bp = gr.Image(label="Raw Channels", type="pil", width=300, height=300)
-            embeddings_img_bp = gr.Image(label="Embeddings", type="pil", width=300, height=300)
+            raw_img_bp = gr.Image(label="Raw Channels", type="pil", width=300, height=500)
+            embeddings_img_bp = gr.Image(label="Embeddings", type="pil", width=300, height=500)
 
         # Update the confusion matrices whenever sliders change
         data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown], outputs=[raw_img_bp, embeddings_img_bp])