Update app.py

app.py

@@ -179,7 +179,8 @@ def inference(image):
   image_path = image
   
   image_tensor = cv2.imread(image_path) 
-  mask = predict(net, image_tensor)
+  with torch.no_grad():
+      mask = predict(net, image_tensor)
   
 
   return [mask,mask]