osanseviero commited on
Commit
fbdd41c
·
1 Parent(s): e61a53a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  os.system("git clone https://github.com/thohemp/6DRepNet")
3
  import sys
4
- sys.path.append("frame-interpolation")
5
 
6
  from model import SixDRepNet
7
  import math
@@ -79,7 +79,7 @@ def predict(img):
79
  r_pred_deg = euler[:, 2].cpu()
80
  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)
81
 
82
- return img
83
 
84
 
85
  iface = gr.Interface(
 
1
  import os
2
  os.system("git clone https://github.com/thohemp/6DRepNet")
3
  import sys
4
+ sys.path.append("6DRepNet")
5
 
6
  from model import SixDRepNet
7
  import math
 
79
  r_pred_deg = euler[:, 2].cpu()
80
  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)
81
 
82
+ return img
83
 
84
 
85
  iface = gr.Interface(