make api

app.py

@@ -33,7 +33,7 @@ def make_transparent_foreground(image, mask):
     return np.where(new_mask, alpha_im, bg).astype(np.uint8)
 
 
-def predict(image):
+def makeMask(image):
     input_tensor = image_to_tensor(image)
     input_batch = input_tensor.unsqueeze(0)  # create a mini-batch as expected by the model
 
@@ -52,8 +52,8 @@ def predict(image):
     return np.where(mask, 255, background).astype(np.uint8)
 
 
-def inference(image):
-    mask = predict(image)
+def predict(image):
+    mask = makeMask(image)
     return make_transparent_foreground(image, mask)
 
 
@@ -79,7 +79,7 @@ This demo is running on a CPU, if you like this project please make us a donatio
 """
 
 demo = gr.Interface(
-    inference, [
+    predict, [
         gr.Image(type="pil", label="Image"),
     ], [
         gr.Image(type="pil", label="Image alpha background")