Update app.py

app.py

@@ -41,9 +41,17 @@ elif opt == 'Upload image via link':
             time.sleep(4)
             show.empty()
 
-if image is not None:
-    st.image(image, width=256, caption='Uploaded Image')
+try:
+  if image is not None:
+    st.image(image, width = 300, 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.')
+        img = preprocess(image)
+
+        model = model_arc()
+        model.load_weights("classify_model.h5")
+
+        prediction = model.predict(img[np.newaxis, ...])
+        st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[np.argmax(prediction[0], axis=-1)]))
+except Exception as e:
+  st.info(e)
+  pass