DSCmatter commited on
Commit
0b4ae93
·
1 Parent(s): 8c52ee1

slight change

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,14 +1,13 @@
1
  import streamlit as st
2
- # import tensorflow as tf # not needed just to infer
3
- from tf_keras.models import load_model
4
- from tensorflow.keras.preprocessing import image
5
  from tensorflow.keras.applications.resnet50 import preprocess_input
6
  import numpy as np
7
  from PIL import Image
8
 
9
  @st.cache_resource
10
  def load_my_model():
11
- return load_model("resnet50_dryfruits.h5", compile=False)
12
 
13
  class_names = {
14
  0: 'AlmondGrade_A',
 
1
  import streamlit as st
2
+ from tf_keras.models import load_model # <-- switch to tf_keras
3
+ from tensorflow.keras.preprocessing import image
 
4
  from tensorflow.keras.applications.resnet50 import preprocess_input
5
  import numpy as np
6
  from PIL import Image
7
 
8
  @st.cache_resource
9
  def load_my_model():
10
+ return load_model("resnet50_dryfruits.h5", compile=False)
11
 
12
  class_names = {
13
  0: 'AlmondGrade_A',