Update app.py

app.py

@@ -6,6 +6,7 @@ import torch
 import json
 from gradio import ChatMessage
 import os
+import matplotlib.pyplot as plt
 
 IN_SPACE = bool(os.environ.get("SPACE_AUTHOR_NAME", False))
 
@@ -279,6 +280,13 @@ def show_chats(idx, df, file, all_check):
                     btn = gr.Button("Clear")
                     btn.click(lambda: gr.Textbox(value=""), None, text)
 
+def update_plot(df, file, all_check):
+    if all_check or file is None or file == "":
+        loss_col = "avg_loss"
+    else:
+        loss_col = f"loss_{exps[file]}"
+
+    return plt.hist(df[loss_col], bins=50)
 
 
 
@@ -326,6 +334,9 @@ with gr.Blocks() as demo:
         with gr.Column():
             chat_b = gr.Chatbot(label="Model B", type="messages", height=1000)
 
+    with gr.Row():
+        plot = gr.Plot()
+
     lower_limit.change(
         filter_df,
         inputs=[lower_limit, upper_limit, file, all_check],
@@ -337,6 +348,8 @@ with gr.Blocks() as demo:
         outputs=[idx, hidden_df],
     )
 
+    hidden_df.change(update_plot, [hidden_df, file, all_check], plot)
+
     idx.change(
         show_chats,
         inputs=[idx, hidden_df, file, all_check],
@@ -345,6 +358,8 @@ with gr.Blocks() as demo:
     prev_btn.click(lambda x: max(0, x - 1), inputs=idx, outputs=idx)
     next_btn.click(lambda x: x + 1, inputs=idx, outputs=idx)
 
+    
+
 
 if __name__ == "__main__":
     demo.launch(debug=True)