added id per job

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 from segmentation import get_mask,replace_sofa
 from styleTransfer import create_styledSofa
 from PIL import Image
+from random import randint
 
 #https://colab.research.google.com/drive/11CtQpSeRBGAuw4TtE_rL470tRo-1X-p2#scrollTo=edGukUHXyymr
 #https://colab.research.google.com/drive/1xq33YKf0LVKCkbbUZIoNPzgpR_4Kd0qL#scrollTo=sPuM8Xypjs-c
@@ -89,7 +90,8 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     Return:
         new_sofa  = image containing the styled sofa
     """
-    
+    id = randint(0, 10)
+    print('Starting job ', id)
     # preprocess input images to be (640,640) squares to fit requirements of the segmentation model
     input_img,style_img = Image.fromarray(input_img),Image.fromarray(style_img)
     resized_img,box = resize_sofa(input_img)
@@ -107,6 +109,7 @@ def style_sofa(input_img: np.ndarray, style_img: np.ndarray):
     print('Replacing sofa...')
     new_sofa = replace_sofa(resized_img,mask,styled_sofa)
     new_sofa = new_sofa.crop(box)
+    print('Finishing job', id)
     return new_sofa
 
 image = gr.inputs.Image()