Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import gradio as gr
 import keras
 import pandas as pd
+import joblib
 
 from transformers import pipeline
 
@@ -29,8 +30,8 @@ def get_murmur_from_recordings(audio):
 
 
 def get_patient_outcome(age, sex, height, weight, is_pregnant, murmur):
-    model = keras.models.load_model('patient_outcome_classifier.keras')
-
+    # model = keras.models.load_model('patient_outcome_classifier.keras')
+    model = joblib.load("patient_outcome_classifier_v3.joblib")
     is_pregnant = 1 if is_pregnant else 0
     sex2int = {'Male': 0, 'Female': 1}
     sex = sex2int[sex]