Update app.py

app.py

@@ -5,16 +5,6 @@ from PIL import Image
 import urllib.request
 from utils import *
 
-def build_and_load_model():
-    # Initialize and build the model first
-    model = model_arc()
-    dummy_input = tf.zeros((1, 256, 256, 3))
-    model.build(input_shape=dummy_input.shape)
-    model(dummy_input)
-    model.load_weights("classify_model.h5")
-    return model
-
-model = build_and_load_model()
 labels = gen_labels()
 
 html_temp = '''
@@ -22,7 +12,6 @@ html_temp = '''
     <center><h1>Garbage Segregation</h1></center>
     </div>
     '''
-
 st.markdown(html_temp, unsafe_allow_html=True)
 
 html_temp = '''
@@ -31,7 +20,6 @@ html_temp = '''
     <center><h3>Please upload Waste Image to find its Category</h3></center>
     </div>
     '''
-
 st.markdown(html_temp, unsafe_allow_html=True)
 
 opt = st.selectbox("How do you want to upload the image for classification?\n", ('Please Select', 'Upload image via link', 'Upload image from device'))
@@ -47,19 +35,15 @@ elif opt == 'Upload image via link':
     try:
         img = st.text_input('Enter the Image Address')
         image = Image.open(urllib.request.urlopen(img)).resize((256, 256), Image.LANCZOS)
-    except:
+    except ValueError:
         if st.button('Submit'):
             show = st.error("Please Enter a valid Image Address!")
             time.sleep(4)
             show.empty()
 
-try:
-    if image is not None:
-        st.image(image, width=256, caption='Uploaded Image')
-        if st.button('Predict'):
-            img = preprocess(image)
-
-            prediction = model.predict(img)
-            st.info('Hey! The uploaded image has been classified as "{} waste" '.format(labels[np.argmax(prediction)]))
-except Exception as e:
-    st.info(str(e))
+if image is not None:
+    st.image(image, width=256, caption='Uploaded Image')
+    if st.button('Predict'):
+        img_array = preprocess(image)
+        prediction = model.predict(img_array)
+        st.info(f'Hey! The uploaded image has been classified as "{labels[np.argmax(prediction)]}" waste.')