Dewasheesh commited on
Commit
9ee4ab4
·
verified ·
1 Parent(s): ffd3dd3

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +1 -0
app.py CHANGED
@@ -24,11 +24,11 @@ tool_wear = st.number_input("Tool Wear (min)", min_value=0, max_value=300, value
24
 
25
  # Assemble input into DataFrame
26
  input_data = pd.DataFrame([{
27
- 'Air temperature': air_temp,
28
- 'Process temperature': process_temp,
29
- 'Rotational speed': rot_speed,
30
  'Torque': torque,
31
- 'Tool wear': tool_wear,
32
  'Type': Type
33
  }])
34
 
 
24
 
25
  # Assemble input into DataFrame
26
  input_data = pd.DataFrame([{
27
+ 'Air_temperature': air_temp,
28
+ 'Process_temperature': process_temp,
29
+ 'Rotational_speed': rot_speed,
30
  'Torque': torque,
31
+ 'Tool_wear': tool_wear,
32
  'Type': Type
33
  }])
34
 
requirements.txt CHANGED
@@ -4,3 +4,4 @@ streamlit==1.43.2
4
  joblib==1.5.1
5
  scikit-learn==1.6.0
6
  xgboost==2.1.4
 
 
4
  joblib==1.5.1
5
  scikit-learn==1.6.0
6
  xgboost==2.1.4
7
+ mlflow==3.0.1