Fluospark128 commited on
Commit
0d35bdf
·
verified ·
1 Parent(s): aefd9a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,6 +1,6 @@
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
@@ -13,7 +13,7 @@ import numpy as np
13
  from PIL import Image
14
 
15
  # Load the model
16
- @st.cache_resource
17
  def load_model():
18
  model = tf.keras.models.load_model("cassava_leaf_disease_model.keras") #tf.keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") cassava_model.h5
19
  return model
@@ -24,8 +24,8 @@ model = load_model()
24
  CLASS_NAMES = ["Cassava Bacterial Blight Disease", "Cassava Brown Streak Disease", "Cassava Green Mottle Disease", "Cassava Mosaic Disease", "Healthy"]
25
 
26
  # Streamlit UI
27
- st.title("Cassava Leaf Disease Classifier")
28
- st.write("Upload an image of a cassava leaf to classify its disease.")
29
 
30
  # File uploader
31
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
 
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
 
13
  from PIL import Image
14
 
15
  # Load the model
16
+ #@st.cache_resource
17
  def load_model():
18
  model = tf.keras.models.load_model("cassava_leaf_disease_model.keras") #tf.keras.saving.load_model("hf://Fluospark128/Cassava_Disease_Classifier") cassava_model.h5
19
  return model
 
24
  CLASS_NAMES = ["Cassava Bacterial Blight Disease", "Cassava Brown Streak Disease", "Cassava Green Mottle Disease", "Cassava Mosaic Disease", "Healthy"]
25
 
26
  # Streamlit UI
27
+ st.title("Cassava Leaf Disease Identifier")
28
+ st.write("Upload an image of a cassava leaf to identify its disease.")
29
 
30
  # File uploader
31
  uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])