Update app.py

app.py

@@ -222,6 +222,7 @@ def run_inference(image):
     # Convert output to image
     output = output.cpu().squeeze(0).permute(1, 2, 0).numpy()
     output = ((output - output.min()) / (output.max() - output.min()) * 255).astype(np.uint8)
+    rp(output)
     return output
     
 def to_hub(model):