Update website.py

website.py

@@ -1,8 +1,28 @@
+from huggingface_hub import hf_hub_download
 import gradio as gr
-#import main
+import tensorflow as tf
+import numpy as np
 from dateutil.utils import today
 
+model_path = hf_hub_download(repo_id="MaxJalo/CardioAI", filename="cardioai_model.keras")
+model = tf.keras.models.load_model(model_path)
 
+
+def webai(user_input):
+    user_input_clear = user_input
+    input_data = [user_input_clear]
+    input_data_scaled = (input_data - X_min) / (X_max - X_min)
+    print(input_data_scaled)
+    # Получаем предсказание от модели
+    predicted_result_scaled = model.predict(input_data_scaled)
+    print(predicted_result_scaled[0][0] * 100)
+    # 35 0 190 75 120 80 1 1 0 0 1
+    # 35 0 170 90 130 90 1 1 0 0 0
+    # 39 0 156 45 110 80 2 1 0 0 0
+    # 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]
@@ -31,8 +51,7 @@ def pomoch(age, gender, height, weight, ap_hi, ap_lo, cholesterol, gluc, smoke,
             X[7] = 2
         else:
             X[7] = 3
-    return X
-#    return main.webai(X)
+return webai(X)
 
 
 demo = gr.Interface(