hotfix

app.py

@@ -48,12 +48,12 @@ def predict(img):
         direction_str = ""
         if p_pred_deg > th:
             direction_str = "UP "
-        elif p_pred_deg < th:
+        elif p_pred_deg < -th:
             direction_str ="DOWN "
         
         if y_pred_deg > th:
             direction_str += "LEFT"
-        elif y_pred_deg < th:
+        elif y_pred_deg < -th:
             direction_str += "RIGHT"
 
         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"