Update app.py

app.py

@@ -10,36 +10,23 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 # Import optimization logic
 from models.optimizer import optimize_design
 
-
 # Load pre-trained model
 def load_model():
-    """
-    Load the pre-trained defect prediction model from the models directory.
-    """
     with open("models/defect_model.pkl", "rb") as file:
         model = pickle.load(file)
     return model
 
-
 # Predict defects using the loaded model
 def predict_defects(model, data):
-    """
-    Predict defect types based on input data using the pre-trained model.
-    """
     predictions = model.predict(data)
     return predictions
 
-
 def main():
-    """
-    Main Streamlit web app for defect prediction and design optimization.
-    """
     st.title("Press Tool AI: Defect Prediction and Optimization")
 
     # File upload
     uploaded_file = st.file_uploader("Upload Design Parameters (CSV)", type="csv")
     if uploaded_file:
-        # Load uploaded CSV data
         data = pd.read_csv(uploaded_file)
         st.write("Uploaded Data:")
         st.dataframe(data)
@@ -66,6 +53,5 @@ def main():
             mime="text/csv",
         )
 
-
 if __name__ == "__main__":
     main()