Update main.py

main.py

@@ -244,18 +244,19 @@ class ProstateSegmentationAlgorithm(SegmentationAlgorithm):
 
 
 def predict(input_file):
+    print("Making prediction")
     print(input_file)
-    return "cat"
+    return input_file
 
 
 demo = gr.Interface(
     fn=predict,
     inputs=gr.File(file_count="single", file_types=[".mha", ".nii.gz", ".nii"]),
-#     outputs=(
-#         gr.File()
-#     )
-# ,
-    outputs=gr.Label(num_top_classes=3),
+    outputs=(
+        gr.File()
+    )
+,
+    # outputs=gr.Label(num_top_classes=3),
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)