Update app.py

app.py

@@ -55,7 +55,8 @@ else:
 
 model.eval()
 
-def segment(inp):
+def segment(input):
+    inp = input
     x = inp.transpose([2, 0, 1])  #  channels-first
     x = np.expand_dims(x, axis=0)  # adding a batch dimension