add type='numpy'

app.py

@@ -34,8 +34,8 @@ gr.Interface(inference,
               gr.inputs.Textbox(placeholder='e.g., cFOS_in_HC_ensemble.pt', label='model_name'),
               gr.inputs.Image(type='numpy', label='Input image')
              ],
-             [gr.outputs.Image(label='Segmentation Mask'),
-              gr.outputs.Image(label='Uncertainty Map')],
+             [gr.outputs.Image(type='numpy', label='Segmentation Mask'),
+              gr.outputs.Image(type='numpy', label='Uncertainty Map')],
              title=title,
              description=description,
              examples=examples,