Update handler.py

handler.py

@@ -1,3 +1,4 @@
+import time
 import torch
 from transformers import CLIPProcessor, CLIPModel
 from PIL import Image
@@ -23,6 +24,8 @@ class EndpointHandler:
 
     def __call__(self, data):
         """Processes input and runs inference."""
+        start_time = time.time()  # Start timer
+
         print("📥 Processing input...")
 
         if "inputs" in data:
@@ -61,4 +64,9 @@ class EndpointHandler:
             best_matches = [(text[idx], probs[idx].item()) for idx in sorted_indices[:3]]  # Get top 3 matches
             predictions.append({"image_index": i, "top_matches": best_matches})
 
-        return {"predictions": predictions}
+        total_time = time.time() - start_time  # Calculate time taken
+
+        return {
+            "predictions": predictions,
+            "processing_time_seconds": round(total_time, 4)
+        }