v1

app.py

@@ -15,7 +15,9 @@ inpaint_model = inpaint.IVModel(device=device)
 sod_model = sod.IVModel(device=torch.device("cpu"))
 max_size=512
 scale_factor = 8
+count = 0
 def sod_inpaint(img):
+    global count
     h,w = img.shape[:2]
     if max(h, w) > max_size:
         if h < w:
@@ -27,16 +29,16 @@ def sod_inpaint(img):
     img = cv2.resize(img, (w,h))
     img = img[:,:,::-1]
     res = sod_model.forward(img,None)
-    print('sod complete!')
     res = np.uint8(res)
     res = inpaint_model.forward(res,None)
     res = np.uint8(res)
-    print('inpaint complete!')
+    count +=1
+    print(count, ' images have been processed')
     return res[:,:,::-1]
 
 
 
 examples = glob.glob('examples/*.*')
 inputs = gr.inputs.Image(shape=None, image_mode="RGB", invert_colors=False, source="upload", tool="editor", type="numpy", label=None, optional=False)
-iface = gr.Interface(fn=sod_inpaint, inputs=inputs, outputs="image", examples=examples, title='Salient Object Detection + Inpaint', description='Upload a image and you will find something disappear', theme='huggingface')
+iface = gr.Interface(fn=sod_inpaint, inputs=inputs, outputs="image", examples=examples, title='Salient Object Detection + Inpaint', description='Upload an image and you will find something disappears', theme='huggingface')
 iface.launch()