osanseviero commited on
Commit
b7d169d
·
1 Parent(s): d3201bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def predict(frame):
66
  r_pred_deg = euler[:, 2].cpu()
67
  arr = utils.plot_pose_cube(frame, y_pred_deg, p_pred_deg, r_pred_deg, x_min + int(.5*(x_max-x_min)), y_min + int(.5*(y_max-y_min)), size = bbox_width)
68
  img = Image.fromarray(arr)
69
- return img.resize(size=(200, 200))
70
 
71
  title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
72
  description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."
 
66
  r_pred_deg = euler[:, 2].cpu()
67
  arr = utils.plot_pose_cube(frame, y_pred_deg, p_pred_deg, r_pred_deg, x_min + int(.5*(x_max-x_min)), y_min + int(.5*(y_max-y_min)), size = bbox_width)
68
  img = Image.fromarray(arr)
69
+ return img.resize(size=(500, 500))
70
 
71
  title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
72
  description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."