Update demo.py

demo.py

@@ -155,7 +155,7 @@ def main():
         top_predictions = [(LABEL_LIST[i], float(score)) for i, score in zip(topk_indices, topk_scores)]
         
         image_name = os.path.basename(image_path)
-        output_str = f"\nTop 3 Predictions:\n"
+        output_str = f"Frame: {image_name.split(".")[0]}\n\nTop 3 Predictions:\n"
         for rank, (lbl, score) in enumerate(top_predictions, start=1):
             output_str += f"Rank {rank}: {lbl} ({score:.4f})\t\t\t"
         print(f"Selected image: {image_name}")