Update app.py

app.py

@@ -51,11 +51,11 @@ try:
     if st.button('Predict'):
         img = preprocess(image)
 
-        model = ('classify_model.h5')
+        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)]))
+        st.info('Hey! The uploaded image has been classified as " {} waste " '.format(labels[prediction]))#np.argmax(prediction[0], axis=-1)]))
 except Exception as e:
   st.info(e)
   pass