Update app.py

app.py

@@ -1,20 +1,20 @@
 import gradio as gr
 import joblib
-import numpy as np  # Ensure numpy is imported
+import numpy as np 
 
 # Load the model and scaler
 model = joblib.load('logistic_regression_model.pkl')
 scaler = joblib.load('scaler.pkl')
 
+# Define risk thresholds globally
+high_risk_threshold = 0.75
+moderate_risk_threshold = 0.25
+
 def initial_risk_check(features):
     # Scale and predict risk based on the features
     features_scaled = scaler.transform([features])
     prediction = model.predict_proba(features_scaled)[:, 1]
     
-    # Define risk thresholds
-    high_risk_threshold = 0.75
-    moderate_risk_threshold = 0.25
-    
     # Determine risk level and recommendation
     if prediction >= high_risk_threshold:
         return "High risk. Please consult a doctor immediately.", prediction