Update website.py

website.py

@@ -1,12 +1,19 @@
 from huggingface_hub import hf_hub_download
 import gradio as gr
+import pandas as pd
 import tensorflow as tf
 import numpy as np
 from dateutil.utils import today
+from datasets import load_dataset
 
 model_path = hf_hub_download(repo_id="MaxJalo/CardioAI", filename="cardioai_model.keras")
 model = tf.keras.models.load_model(model_path)
 
+heart = load_dataset("MaxJalo/CardioAI", split = 'train')
+data = pd.DataFrame(heart, columns=["age","gender","height","weight","ap_hi","ap_lo","cholesterol","gluc","smoke","alco","active",'cardio'])
+X_for_train = data.drop(['cardio'], axis=1).values
+X_min = np.min(X_for_train, axis=0)
+X_max = np.max(X_for_train, axis=0)
 
 def webai(user_input):
     user_input_clear = user_input
@@ -22,7 +29,8 @@ def webai(user_input):
     # 47 1 168 87 120 80 2 1 1 1 1
     # 37 0 185 75 120 80 2 1 1 1 0
     return f"{round(predicted_result_scaled[0][0] * 100, 2)}%"
-    
+
+
 def pomoch(age, gender, height, weight, ap_hi, ap_lo, cholesterol, gluc, smoke, alco, active):
     X = [int(age), gender, int(height), int(weight), int(ap_hi), int(ap_lo), float(cholesterol), float(gluc), smoke,
          alco, active]