Update main.py

main.py

@@ -245,7 +245,9 @@ class ProstateSegmentationAlgorithm(SegmentationAlgorithm):
 
 def predict(input_file):
     print("Making prediction")
-    print(input_file)
+    image = sitk.ReadImage(input_file)
+    print(image.size)
+    ProstateSegmentationAlgorithm().process()
     return input_file
 
 print("Starting interface")