process time

app.py

@@ -6,6 +6,7 @@ from decord import VideoReader, cpu
 import base64
 import io
 import spaces
+import time
 
 # Load model
 model_path = 'openbmb/MiniCPM-V-2_6'
@@ -44,6 +45,8 @@ def encode_video(video_path):
 
 @spaces.GPU
 def analyze_video(prompt, video):
+    start_time = time.time()
+    
     if isinstance(video, str):
         video_path = video
     else:
@@ -69,7 +72,10 @@ def analyze_video(prompt, video):
     
     response = model.chat(image=None, msgs=context, tokenizer=tokenizer, **params)
     
-    return response
+    end_time = time.time()
+    processing_time = end_time - start_time
+    
+    return f"Analysis Result:\n{response}\n\nProcessing Time: {processing_time:.2f} seconds"
 
 with gr.Blocks() as demo:
     gr.Markdown("# Video Analyzer")
@@ -78,7 +84,7 @@ with gr.Blocks() as demo:
             prompt_input = gr.Textbox(label="Prompt")
             video_input = gr.Video(label="Upload Video")
         with gr.Column():
-            output = gr.Textbox(label="Analysis Result")
+            output = gr.Textbox(label="Analysis Result and Processing Time")
     
     analyze_button = gr.Button("Analyze Video")
     analyze_button.click(fn=analyze_video, inputs=[prompt_input, video_input], outputs=output)