Update app.py

app.py

@@ -41,7 +41,7 @@ def greet(image):
     with torch.no_grad():
         mask_pred = model(image)
         
-    return mask_pred[0]
+    return mask_pred[0].numpy()
     
 
 demo = gr.Interface(
@@ -59,7 +59,7 @@ demo = gr.Interface(
     outputs=[
         gr.Image(
             label="Model Prediction",
-            image_mode="RGB",
+            image_mode="L",
             # height=400,
             # width=400,
         )