Spaces:

harshin
/

STREAMLIT_CLASSIFIER

Sleeping

harshin commited on Dec 8, 2023

Commit

3266ddc

1 Parent(s): 1fee3e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ elif task == "Tumor Detection":
     if uploaded_file is not None:
         # Load the tumor detection model
-        model = load_model('CN.keras')
         st.image(uploaded_file, caption="Uploaded Image.", use_column_width=False, width=200)
         st.write("")

     if uploaded_file is not None:
         # Load the tumor detection model
+        model = load_model('CN.h5')
         st.image(uploaded_file, caption="Uploaded Image.", use_column_width=False, width=200)
         st.write("")