add unit descriptor in the predictor

zengyh1900 · Feb 23, 2020 · 6c20fb9 · 6c20fb9
1 parent 0f8816c
commit 6c20fb9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/demo.py b/src/demo.py
@@ -144,7 +144,10 @@ def set_dialog(self):
       edit = QtWidgets.QLineEdit()
       self.editList.append(edit)
       label = QtWidgets.QLabel()
-      label.setText(utils.M_STR[i])
+      if i == 0:
+        label.setText('{} (x2 kg)'.format(utils.M_STR[i]))
+      else:
+        label.setText('{} (cm)'.format(utils.M_STR[i]))
       # label.setFont(QFont("Arial", 11, QFont.Bold))
       label.setFont(QFont("Arial", 12))
       label.setFixedHeight(20)