jhjh

App.py

@@ -21,7 +21,10 @@ if uploaded_file is not None:
         return Draw(image)
     
     # Process the image and retrieve results
-    image, Results = process_image(uploaded_file)
+    image, Results,execution_time = process_image(uploaded_file)
+    # Execution Time
+    st.write(f"Execution Time: {execution_time:.2f} seconds")
+    # Change Image
     Change_Image(image,image_initiale)
     # Some Initializations
     sauvgarder_button = st.sidebar.empty()

utilitis.py

@@ -4,6 +4,7 @@ from PIL import ImageDraw, ImageFont,ImageEnhance
 import torch
 from transformers import AutoProcessor,LayoutLMv3ForTokenClassification
 import numpy as np
+import time
 
 model_Hugging_path = "Noureddinesa/Output_LayoutLMv3_v6"
 
@@ -113,6 +114,8 @@ def Get_Json(true_predictions,words):
 #############################################################################
 #############################################################################   
 def Draw(image):
+    start_time = time.time()
+    
     image = enhance_image(image,1.3,1.5)
     true_predictions, true_boxes,words = Run_model(image)
     draw = ImageDraw.Draw(image)
@@ -155,7 +158,10 @@ def Draw(image):
     # Get the Results Json File 
     Results = Get_Json(true_predictions,words)
     
-    return image,Results
+    end_time = time.time() 
+    execution_time = end_time - start_time 
+    
+    return image,Results,execution_time
 
 #############################################################################
 #############################################################################