Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import os
 import matplotlib.pyplot as plt
 from io import BytesIO
+import base64
 
 # Check if CUDA is available, otherwise use CPU
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -99,7 +100,7 @@ def create_heatmap(frame_scores):
     buf.seek(0)
     plt.close()
     
-    return buf
+    return base64.b64encode(buf.getvalue()).decode('utf-8')
 
 def load_sample_frame(video_path):
     cap = cv2.VideoCapture(video_path)