Update app.py

app.py

@@ -9,6 +9,7 @@ import cv2
 from PIL import Image, ImageDraw
 import insightface
 from insightface.app import FaceAnalysis
+import time
 
 # Diffusion
 model_base = "runwayml/stable-diffusion-v1-5"
@@ -55,6 +56,7 @@ def face_swap(src_img, dest_img):
 
 @spaces.GPU(enable_queue=True)
 def greet(description,color,features,occasion,type_,face):
+    start = time.time()
 
     # Parse input
     prompt = ''
@@ -80,6 +82,9 @@ def greet(description,color,features,occasion,type_,face):
 
     if(face != 'Normal'):
         image = face_swap(face, image)
+
+    end = time.time()
+    print('time:', end - start)
     
     return image