Update app.py

app.py

@@ -43,5 +43,5 @@ def predict(img):
    mask=np.reshape(mask,(480,640))
    
    return Image.fromarray(mask.astype('uint8'))
-   
+  
 gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(128,128)), outputs=gr.inputs.Image(), examples=['color_157.jpg','color_158.jpg']).launch(share=False)