Update app.py

app.py

@@ -67,8 +67,8 @@ def process_video(video_path, target, progress=gr.Progress()):
             break
 
         # Resize the frame
-        img_resized = cv2.resize(img, (640, 360))
-        pil_img = Image.fromarray(cv2.cvtColor(img_resized, cv2.COLOR_BGR2RGB))
+        #img_resized = cv2.resize(img, (1280, 720))
+        pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
         
         batch_frames.append(pil_img)
         batch_indices.append(i)
@@ -127,18 +127,17 @@ def process_video(video_path, target, progress=gr.Progress()):
 def create_heatmap(frame_scores, current_frame):
     plt.figure(figsize=(12, 3))
     plt.imshow([frame_scores], cmap='hot_r', aspect='auto')
-    cbar = plt.colorbar(label='Confidence')
-    cbar.ax.yaxis.set_ticks_position('left')
-    cbar.ax.yaxis.set_label_position('left')
     plt.title('Object Detection Heatmap')
     plt.xlabel('Frame')
     plt.yticks([])
 
+    # Add more frame numbers on x-axis
     num_frames = len(frame_scores)
-    step = max(1, num_frames // 10)
+    step = max(1, num_frames // 10)  # Show at most 10 frame numbers
     frame_numbers = range(0, num_frames, step)
     plt.xticks(frame_numbers, [str(i) for i in frame_numbers])
 
+    # Add vertical line for current frame
     plt.axvline(x=current_frame, color='blue', linestyle='--', linewidth=2)
 
     plt.tight_layout()