Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 from PIL import Image
 from transformers import BlipProcessor, BlipForConditionalGeneration
+import time
 
 processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-large")
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-large")
@@ -15,7 +16,12 @@ def caption(img, min_len, max_len):
     return processor.decode(out[0], skip_special_tokens=True)
 
 def greet(img, min_len, max_len):
-    return caption(img, min_len, max_len)
+    start = time.time()
+    result = caption(img, min_len, max_len)
+    end = time.time()
+    total_time = str(end - start)
+    result = result + '\n' + total_time + ' seconds'
+    return result
 
 iface = gr.Interface(fn=greet, 
                      title='Blip Image Captioning Large',