Update app.py

app.py

@@ -64,13 +64,23 @@ def predict(frame):
     y_pred_deg = euler[:, 1].cpu()
     r_pred_deg = euler[:, 2].cpu()
     return 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)
-  
+ 
+title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
+description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."
+article = "<div style='text-align: center;'><a href='https://github.com/thohemp/6DRepNet' target='_blank'>Github Repo</a> | <center><a href='https://arxiv.org/abs/2202.12555' target='_blank'>Paper</a> |</center></div>"
+
   
 iface = gr.Interface(
     fn=predict, 
     inputs=gr.inputs.Image(label="Input Image", source="webcam"),
     outputs='image',
-    live=True
+    live=True,
+    title=title,
+    description=description,
+    article=article,
 )
+)
+
+iface.launch()
 
 iface.launch()