Dodanie modelu Keras

README.md

@@ -108,8 +108,14 @@ model.fit(
 import numpy as np
 import pandas as pd
 import tensorflow as tf
+from huggingface_hub import hf_hub_download
 
-model = tf.keras.models.load_model("iris_mlp.keras")
+filename = "iris_mlp.keras"
+repo_id = "studentscolab/iris_keras"
+
+model_path = hf_hub_download(repo_id=repo_id, filename=filename)
+
+model = tf.keras.models.load_model(model_path)
 
 x_new = pd.DataFrame([{
     "sepal length (cm)": 5.1,
@@ -123,4 +129,5 @@ pred  = int(np.argmax(proba))
 
 print("Probabilities:", proba)
 print("Predicted class:", pred)
+
 ```