Update app.py

app.py

@@ -24,8 +24,8 @@ import tensorflow as tf
 print(torch.__version__)
 print(torch.version.cuda)
 
-matplotlib.rcParams['figure.dpi'] = 400
-matplotlib.rcParams['savefig.dpi'] = 400
+matplotlib.rcParams['figure.dpi'] = 500
+matplotlib.rcParams['savefig.dpi'] = 500
 
 # Initialize models and other global variables
 device = 'cuda'
@@ -356,7 +356,7 @@ def determine_anomalies(mse_values, threshold=5):
     return anomalies
 
 
-def plot_mse(df, mse_values, title, color='blue', time_threshold=1, hide_first_n=5):
+def plot_mse(df, mse_values, title, color='blue', time_threshold=1, hide_first_n=2):
     plt.figure(figsize=(16, 8), dpi=300)
     fig, ax = plt.subplots(figsize=(16, 8))
 
@@ -514,9 +514,9 @@ def process_video(video_path, desired_fps, batch_size, progress=gr.Progress()):
                 X, feature_columns, raw_embedding_columns, batch_size=batch_size)
 
             progress(0.95, "Generating plots")
-            mse_plot_all = plot_mse(df, mse_all, "Facial Features + Emotions", color='blue', hide_first_n=5)
-            mse_plot_comp = plot_mse(df, mse_comp, "Facial Features", color='deepskyblue', hide_first_n=5)
-            mse_plot_raw = plot_mse(df, mse_raw, "Facial Embeddings", color='steelblue', hide_first_n=5)
+            mse_plot_all = plot_mse(df, mse_all, "Facial Features + Emotions", color='blue', hide_first_n=2)
+            mse_plot_comp = plot_mse(df, mse_comp, "Facial Features", color='deepskyblue', hide_first_n=2)
+            mse_plot_raw = plot_mse(df, mse_raw, "Facial Embeddings", color='steelblue', hide_first_n=2)
 
             emotion_plots = [
                 plot_mse(df, embedding_anomaly_detection(df[emotion].values.reshape(-1, 1)),
@@ -548,8 +548,8 @@ def process_video(video_path, desired_fps, batch_size, progress=gr.Progress()):
 
 # Define gallery outputs
 gallery_outputs = [
-    gr.Gallery(label="Most Frequent Person Random Samples", columns=5, rows=2, height="auto"),
-    gr.Gallery(label="Other Persons Random Samples", columns=5, rows=1, height="auto")
+    gr.Gallery(label="Most Frequent Person Random Samples", columns=10, rows=2, height="auto"),
+    gr.Gallery(label="Other Persons Random Samples", columns=10, rows=1, height="auto")
 ]
 
 # Update the Gradio interface
@@ -558,7 +558,7 @@ iface = gr.Interface(
     inputs=[
         gr.Video(),
         gr.Slider(minimum=1, maximum=20, step=1, value=10, label="Desired FPS"),
-        gr.Slider(minimum=1, maximum=32, step=1, value=8, label="Batch Size")
+        gr.Slider(minimum=1, maximum=32, step=1, value=10, label="Batch Size")
     ],
     outputs=[
         gr.Textbox(label="Anomaly Detection Results"),