UCS2014 commited on
Commit
95d6fb4
·
verified ·
1 Parent(s): 82f0ae2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -24,16 +24,15 @@ TAGLINE = "Real-Time Upper/Lower Mud Weight (MW) Limits For Safe Drilling"
24
 
25
  # Defaults (can be overridden by metas or model.feature_names_in_)
26
  FEATURES_DEFAULT = [
27
- "WOB (klbf)",
28
- "Torque (kft.lbf)",
29
  "SPP (psi)",
30
- "RPM (1/min)",
31
- "ROP (ft/h)",
32
- "Flow Rate (gpm)",
33
  ]
34
 
35
- TARGET_BO_DEFAULT = "Breakout MW"
36
- TARGET_BD_DEFAULT = "Breakdown MW"
37
  PRED_BO = "BO_Pred"
38
  PRED_BD = "BD_Pred"
39
 
@@ -43,9 +42,9 @@ X_UNITS = None
43
  # Model/meta discovery (supports uploads with spaces like "bo_model (1).joblib")
44
  MODELS_DIR = Path("models")
45
  ALT_MODELS_DIR = Path("/mnt/data")
46
- BO_MODEL_CANDIDATES = ["bo_model.joblib", "bo_model (1).joblib", "BO.joblib"]
47
  BD_MODEL_CANDIDATES = ["bd_model.joblib", "BD.joblib"]
48
- BO_META_CANDIDATES = ["bo_meta.json", "bo_meta (1).json"]
49
  BD_META_CANDIDATES = ["bd_meta.json"]
50
 
51
  COLORS = {
 
24
 
25
  # Defaults (can be overridden by metas or model.feature_names_in_)
26
  FEATURES_DEFAULT = [
27
+ "Q (gpm)",
 
28
  "SPP (psi)",
29
+ "T (kft.lbf)",
30
+ "WOB (klbf)",
31
+ "ROP (ft/h)"
32
  ]
33
 
34
+ TARGET_BO_DEFAULT = "BO_Actual"
35
+ TARGET_BD_DEFAULT = "BD_Actual"
36
  PRED_BO = "BO_Pred"
37
  PRED_BD = "BD_Pred"
38
 
 
42
  # Model/meta discovery (supports uploads with spaces like "bo_model (1).joblib")
43
  MODELS_DIR = Path("models")
44
  ALT_MODELS_DIR = Path("/mnt/data")
45
+ BO_MODEL_CANDIDATES = ["bo_model.joblib", "bo_model.joblib", "BO.joblib"]
46
  BD_MODEL_CANDIDATES = ["bd_model.joblib", "BD.joblib"]
47
+ BO_META_CANDIDATES = ["bo_meta.json", "bo_meta.json"]
48
  BD_META_CANDIDATES = ["bd_meta.json"]
49
 
50
  COLORS = {