Update app.py

app.py

@@ -220,11 +220,11 @@ interface = gr.Interface(
     inputs=gr.File(label="Upload FLAIR Image (.nii.gz)"),
     outputs=[
         gr.File(label="Download Segmentation Mask"),
-        gr.Image(label="Input Middle Slice"),
-        gr.Image(label="Output Middle Slice")
+        gr.Image(label="Input: FLAIR image"),
+        gr.Image(label="Output: Lesion Mask")
     ],
     title="FLAMeS: Multiple Sclerosis Lesion Segmentation",
-    description="Upload a skull-stripped FLAIR image (.nii.gz) to generate a binary segmentation of MS lesions."
+    description="Upload a skull-stripped FLAIR image (.nii.gz) to generate a binary segmentation of multiple sclerosis lesions."
 )
 
 # Debugging GPU environment