Fluospark128 commited on
Commit
698fe5b
·
verified ·
1 Parent(s): c9f4ef0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  # Available backend options are: "jax", "torch", "tensorflow".
2
  import os
3
  os.environ["KERAS_BACKEND"] = "tensorflow"
4
- pip install -U keras huggingface_hub
5
 
6
  import keras
7
 
@@ -15,7 +15,7 @@ from PIL import Image
15
  # Load the model
16
  @st.cache_resource
17
  def load_model():
18
- model = keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") #model = tf.keras.models.load_model("cassava_model.h5")
19
  return model
20
 
21
  model = load_model()
 
1
  # Available backend options are: "jax", "torch", "tensorflow".
2
  import os
3
  os.environ["KERAS_BACKEND"] = "tensorflow"
4
+ #pip install -U keras huggingface_hub
5
 
6
  import keras
7
 
 
15
  # Load the model
16
  @st.cache_resource
17
  def load_model():
18
+ model = tf.keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") #model = tf.keras.models.load_model("cassava_model.h5")
19
  return model
20
 
21
  model = load_model()