make service

app.py

@@ -30,7 +30,7 @@ def makeOverlay(image, mask):
     return overlay, colormap
 
 
-def predict(image):
+def makeMask(image):
     with torch.no_grad():
         image = image.resize((512, 512), Image.BILINEAR)
         input_tensor = image_to_tensor(image)
@@ -41,7 +41,7 @@ def predict(image):
         return output.squeeze(0).cpu().numpy().argmax(0)
 
 
-def inference(image):
+def predict(image):
     mask = predict(image)
     overlay, colormap = makeOverlay(image, mask)
     return overlay
@@ -67,7 +67,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="Input"),
     ], [
         gr.Image(type="numpy", label="Image face parsed")