bienom commited on
Commit
d2d0403
1 Parent(s): b3fa822
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -48,12 +48,12 @@ def predict(img):
48
  direction_str = ""
49
  if p_pred_deg > th:
50
  direction_str = "UP "
51
- elif p_pred_deg < th:
52
  direction_str ="DOWN "
53
 
54
  if y_pred_deg > th:
55
  direction_str += "LEFT"
56
- elif y_pred_deg < th:
57
  direction_str += "RIGHT"
58
 
59
  return f"Yaw: {y_pred_deg:0.1f} \n Pitch: {p_pred_deg:0.1f}\n Direction: {direction_str} \n Time: {timemilis:0.2f}ms"
 
48
  direction_str = ""
49
  if p_pred_deg > th:
50
  direction_str = "UP "
51
+ elif p_pred_deg < -th:
52
  direction_str ="DOWN "
53
 
54
  if y_pred_deg > th:
55
  direction_str += "LEFT"
56
+ elif y_pred_deg < -th:
57
  direction_str += "RIGHT"
58
 
59
  return f"Yaw: {y_pred_deg:0.1f} \n Pitch: {p_pred_deg:0.1f}\n Direction: {direction_str} \n Time: {timemilis:0.2f}ms"