Update app.py
Browse files
app.py
CHANGED
@@ -43,5 +43,5 @@ def predict(img):
|
|
43 |
mask=np.reshape(mask,(480,640))
|
44 |
|
45 |
return Image.fromarray(mask.astype('uint8'))
|
46 |
-
|
47 |
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)
|
|
|
43 |
mask=np.reshape(mask,(480,640))
|
44 |
|
45 |
return Image.fromarray(mask.astype('uint8'))
|
46 |
+
|
47 |
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)
|