Update app.py

app.py

@@ -62,11 +62,8 @@ def do_prediction(img):
             label_p_pred = model.predict(img_patch.reshape(1, img_patch.shape[0], img_patch.shape[1], img_patch.shape[2]),
                                          verbose=0)
 
-            print(label_p_pred.shape,'label_p_pred')
-            print(img_patch.shape, 'img_patch')
             seg = np.argmax(label_p_pred, axis=3)[0]
 
-            print(seg.shape, 'seg')
             seg_color = np.repeat(seg[:, :, np.newaxis], 3, axis=2)
 
             if i == 0 and j == 0:
@@ -117,9 +114,6 @@ def do_prediction(img):
 
     prediction_true = prediction_true.astype(np.uint8)
 
-            
-            
-    print(prediction_true.shape, np.unique(prediction_true))  
                 
 
 
@@ -135,7 +129,7 @@ def do_prediction(img):
     print(prediction.shape)
     
     '''
-    return prediction_true
+    return prediction_true * 255
 
 iface = gr.Interface(fn=do_prediction, inputs=gr.Image(), outputs=gr.Image())
 iface.launch()