Update

app.py

@@ -1,8 +1,9 @@
+import multiprocessing
 import os
+import time
 from typing import List, Tuple
-import multiprocessing
-import cv2
 
+import cv2
 import numpy as np
 import pandas as pd
 import streamlit as st
@@ -104,7 +105,7 @@ def inference(file_path: str):
     results: List[Tuple[str, float]] = []
     for index, value in zip(indices, values):
         predicted_label = model.config.id2label[index]
-        print(f"Label: {predicted_label} - {value:.2f}%")
+        # print(f"Label: {predicted_label} - {value:.2f}%")
         results.append((predicted_label, value))
 
     return pd.DataFrame(results, columns=("Label", "Confidence"))
@@ -138,12 +139,16 @@ feature_extractor, model = load_model(model_name)
 VIDEO_TMP_PATH = os.path.join("tmp", "tmp.mp4")
 uploadedfile = st.file_uploader("Upload file", type=["mp4"])
 
+
 if uploadedfile is not None:
     with st.spinner():
         with open(VIDEO_TMP_PATH, "wb") as f:
             f.write(uploadedfile.getbuffer())
 
+    start_time = time.time()
     with st.spinner("Processing..."):
         df = inference(VIDEO_TMP_PATH)
+    end_time = time.time()
+    st.info(f"{end_time - start_time} seconds")
     st.dataframe(df)
     st.video(VIDEO_TMP_PATH)