Update app.py

app.py

@@ -19,25 +19,26 @@ poly = PolynomialFeatures(degree=5, include_bias=False)
 scaler = StandardScaler()
 
 # 
-def preprocess_input(flange_width, beam_width, geometric_factor):
+def preprocess_input(eometric_factor, flange_width, beam_width, beamh, columntw):
     """
     Preprocess the input data: apply polynomial features and then scale.
     """
     
     # # Apply Polynomial Features
-    X_test_poly = poly_loaded.transform([[flange_width, beam_width, geometric_factor]])
+    # ['GEOMETRIC_FACTOR',	'COLUMN_WIDTH',	'BEAM_WIDTH','BEAM - h', 'COLUMN - tw']
+    X_test_poly = poly_loaded.transform([[eometric_factor, flange_width, beam_width, beamh, columntw]])
 
     # Standardize the polynomial features
     scaled_features = scaler_loaded.transform(X_test_poly)
     
     return scaled_features
 
-def predict_fatigue_crack_growth_rate(flange_width, beam_width, geometric_factor):
+def predict_fatigue_crack_growth_rate(geometric_factor, flange_width, beam_width, beamh, columntw):
     """
     Function to predict the fatigue crack growth rate based on the preprocessed input features.
     """
     # Preprocess the input data
-    preprocessed_input = preprocess_input(flange_width, beam_width, geometric_factor)
+    preprocessed_input = preprocess_input(eometric_factor, flange_width, beam_width, beamh, columntw)
     
     # Predict using the loaded model
     prediction = model.predict(preprocessed_input)
@@ -47,12 +48,14 @@ def predict_fatigue_crack_growth_rate(flange_width, beam_width, geometric_factor
 
 # Create a Gradio interface
 iface = gr.Interface(fn=predict_fatigue_crack_growth_rate,
-                     inputs=[gr.Number(label="Column Width"),
+                     inputs=[gr.Number(label="Geometric Factor")],
+                             gr.Number(label="Column Width"),
                              gr.Number(label="Beam Width"),
-                             gr.Number(label="Geometric Factor")],
+                             gr.Number(label="Beam - h"),
+                             gr.Number(label="Column - tw"),
                      outputs=gr.Textbox(label="Prediction"),
                      title="Fatigue Crack Growth Rate Predictor",
-                     description="Enter the values for Column Width, Beam Width, and Geometric Factor to predict the Fatigue Crack Growth Rate.")
+                     description="Enter the values for Geometric Factor, Column Width, Beam Width, Beam - h, and COLUMN - tw to predict the Fatigue Crack Growth Rate.")
 
 if __name__ == "__main__":
     iface.launch()