Update gradio_utils/utils.py

gradio_utils/utils.py

@@ -194,12 +194,13 @@ def process(query_img, state,
     load_checkpoint(model, checkpoint_path, map_location='cpu')
     model.eval()
     outputs = estimate(model, data)
-	
+	print("Output Avail!")
     # visualize results
     vis_s_weight = target_weight_s[0]
     vis_s_image = support_img[0].detach().cpu().numpy().transpose(1, 2, 0)
     vis_q_image = q_img[0].detach().cpu().numpy().transpose(1, 2, 0)
     support_kp = kp_src_3d
+	print("Try to Plot")
     out = plot_results(vis_s_image,
                        vis_q_image,
                        support_kp,
@@ -211,6 +212,7 @@ def process(query_img, state,
                        original_skeleton=state['skeleton'],
                        img_alpha=1.0,
                        )
+    print("Plot Success!")
     return out