Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -5
src/streamlit_app.py
CHANGED
|
@@ -13,14 +13,10 @@ CLASS_NAMES = ["Arborio", "Basmati", "Ipsala", "Jasmine", "Karacadag"]
|
|
| 13 |
|
| 14 |
@st.cache_resource
|
| 15 |
def load_cached_model():
|
| 16 |
-
return load_model(
|
| 17 |
|
| 18 |
model = load_cached_model()
|
| 19 |
|
| 20 |
-
"""@st.cache_resource
|
| 21 |
-
def load_model():
|
| 22 |
-
model = tf.keras.models.load_model(MODEL_PATH)
|
| 23 |
-
return model"""
|
| 24 |
|
| 25 |
def preprocess(File):
|
| 26 |
img = Image.open(File).convert("RGB")
|
|
|
|
| 13 |
|
| 14 |
@st.cache_resource
|
| 15 |
def load_cached_model():
|
| 16 |
+
return load_model(MODEL_PATH)
|
| 17 |
|
| 18 |
model = load_cached_model()
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
def preprocess(File):
|
| 22 |
img = Image.open(File).convert("RGB")
|