Update main.py

main.py

@@ -6,7 +6,11 @@ def predict(img):
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Image(type="pil"),
+    inputs=gr.File(file_count="single", file_types=[".mha", ".nii.gz", ".nii"]),
+#     outputs=(
+#         gr.File()
+#     )
+# ,
     outputs=gr.Label(num_top_classes=3),
 )