Update app.py

app.py

@@ -1,21 +1,16 @@
 import streamlit as st
 from datasets import load_dataset, Image
-from huggingface_hub import hf_hub_download
-pip install joblib
-import joblib
-
-REPO_ID = "jableable/road_project"
-FILENAME = "best_model.keras"
-
-model = joblib.load(
-    hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
-)
+from huggingface_hub import from_pretrained_keras
 
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
 
-loaded_model = keras.saving.load_model("best_model.keras")
+#loaded_model = keras.saving.load_model("best_model.keras")
 
+model = from_pretrained_keras("best_model.keras")
+prediction = model.predict(image)
+prediction = tf.squeeze(tf.round(prediction))
+print(f'The image is a {classes[(np.argmax(prediction))]}!')
 
 
 dataset = load_dataset("beans", split="train")