Spaces:

EcoClim-Solution
/

EcoIdentify

Running

VishnuEcoClim commited on Dec 31, 2023

Commit

91e61f6

1 Parent(s): f434ef0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,7 @@ try:
         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)]))

         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)]))